anndata.experimental.AnnCollection.iterate_axis

anndata.experimental.AnnCollection.iterate_axis#

AnnCollection.iterate_axis(batch_size, axis=0, shuffle=False, drop_last=False)[source]#

沿着一个轴迭代惰性对象。

Parameters:
batch_size int

迭代时每个批次要放多少样本。

axis Literal[0, 1] (default: 0)

要迭代的轴。

shuffle bool (default: False)

设置为 True 以在迭代之前重新排列索引。

drop_last bool (default: False)

设置为 True 以丢弃长度低于 batch_size 的批次。