Method batch

graphlearn/python/gsl/dag_node.py:169–175

(self, batch_size)

    return self

  def batch(self, batch_size):
    assert isinstance(batch_size, int) and batch_size > 0
    self._shape = (batch_size,)
    self._add_param(pywrap.kBatchSize, batch_size)
    self._add_param(pywrap.kEpoch, sys.maxsize >> 32)
    self._add_param(pywrap.kStrategy, "by_order")
    return self

  def shuffle(self, traverse=False):
    strategy_map = {True: "shuffle", False: "random"}