mindspore.dataset.DSCallback
- class mindspore.dataset.DSCallback(step_size=1)[源代码]
数据处理回调类的抽象基类,用户可以基于此类实现自己的回调操作。
用户可通过 ds_run_context 获取数据处理管道相关信息,包括 cur_epoch_num (当前epoch数)、 cur_step_num_in_epoch (当前epoch的step数)、 cur_step_num (当前step数)。
参数:
step_size (int, optional) - 定义相邻的 ds_step_begin/ds_step_end 调用之间相隔的step数,默认值:1,表示每个step都会调用。
样例:
>>> from mindspore.dataset import DSCallback >>> from mindspore.dataset.transforms import transforms >>> >>> class PrintInfo(DSCallback): ... def ds_epoch_end(self, ds_run_context): ... print(ds_run_context.cur_epoch_num) ... print(ds_run_context.cur_step_num) >>> >>> dataset = ds.MnistDataset(mnist_dataset_dir, num_samples=100) >>> op = transforms.OneHot(10) >>> dataset = dataset.map(operations=op, callbacks=PrintInfo())
- ds_epoch_begin(ds_run_context)[源代码]
用于定义在每个数据epoch开始前执行的回调方法。
参数:
ds_run_context (RunContext) - 数据处理管道运行信息。
- ds_epoch_end(ds_run_context)[源代码]
用于定义在每个数据epoch结束后执行的回调方法。
参数:
ds_run_context (RunContext) - 数据处理管道运行信息。