def get_minibatch_iterator(seed=8675309, dataorderseed=0, nBatch=10, nObsBatch=None, nObsTotal=25000, nLap=1, **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) DataIterator = MinibatchIterator(Data, nBatch=nBatch, nObsBatch=nObsBatch, nLap=nLap, dataseed=seed) DataIterator.summary = get_data_info() return DataIterator
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