TrainDataSampler

class pybind11_ke.data.TrainDataSampler(nbatches: int, sampler: Callable[[], dict[str, Union[numpy.ndarray, str]]])

将 pybind11_ke.data.TrainDataLoader.sampling() 或 pybind11_ke.data.TrainDataLoader.cross_sampling() 进行封装。

__init__(nbatches: int, sampler: Callable[[], dict[str, Union[numpy.ndarray, str]]])

创建 TrainDataSample 对象。

参数:

• nbatches (int) – 1 epoch 有多少个 batch

• sampler (pybind11_ke.data.TrainDataLoader.sampling() 或 pybind11_ke.data.TrainDataLoader.cross_sampling()) – 采样器

__iter__()

迭代器函数 iterator.__iter__()

__len__() → int

len() 要求 object.__len__()

返回:

nbatches

返回类型:

int

__next__() → dict[str, Union[numpy.ndarray, str]]

迭代器函数 iterator.__next__()

返回:

采样一批数据

返回类型:

dict[str, Union[np.ndarray, str]]

__weakref__

list of weak references to the object (if defined)

nbatches: int

1 epoch 有多少个 batch

sampler: Callable[[], dict[str, Union[numpy.ndarray, str]]]

pybind11_ke.data.TrainDataLoader.sampling() 或 pybind11_ke.data.TrainDataLoader.cross_sampling() 函数