def get_minibatch_iterator(seed=8675309, dataorderseed=0, nBatch=10, nObsBatch=None, nObsTotal=25000, nLap=1, startLap=0, **kwargs): ''' Args -------- seed : integer seed for random number generator, used for actually *generating* the data dataorderseed : integer seed that determines (a) how data is divided into minibatches (b) order these minibatches are traversed Returns ------- bnpy MinibatchIterator object, with nObsTotal observations divided into nBatch batches ''' X, TrueZ = get_X(seed, nObsTotal) Data = XData(X=X) Data.summary = get_data_info() DataIterator = MinibatchIterator(Data, nBatch=nBatch, nObsBatch=nObsBatch, nLap=nLap, startLap=startLap, dataorderseed=dataorderseed) return DataIterator
def setUp(self): X = np.random.randn(100, 3) self.Data = XData(X=X) self.DataIterator = MinibatchIterator(self.Data, nBatch=10, nLap=10)
def get_minibatch_iterator(seed=8675309, nObsTotal=25000, **kwargs): X, TrueZ = generateData(seed, nObsTotal) Data = XData(X=X, TrueZ=TrueZ) DataIterator = MinibatchIterator(Data, **kwargs) DataIterator.summary = get_data_info() return DataIterator