Python mxnet.module.BaseModule.iter_predict用法及代码示例

用法: iter_predict(eval_data, num_batch=None, reset=True, sparse_row_id_fn=None)

参数：

eval_data：(DataIter) - 用于运行预测的评估数据。
num_batch：(int) - 默认为None，表示运行数据迭代器中的所有批次。
reset：(bool) - 默认为True，指示我们是否应该在开始进行预测之前重置数据迭代器。
sparse_row_id_fn：(A callback function) - 函数需要data_batch作为输入并返回 str -> NDArray 的字典。生成的 dict 用于从 kvstore 中提取 row_sparse 参数，其中 str 键是参数的名称，值是要提取的参数的行 ID。

迭代预测。

例子：

>>> for pred, i_batch, batch in module.iter_predict(eval_data):
...     # pred is a list of outputs from the module
...     # i_batch is a integer
...     # batch is the data batch from the data iterator

相关用法

注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.module.BaseModule.iter_predict。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。