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
