Example #1
0
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
Example #2
0
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