在mxnet上运行预测时出错:检查失败:i> = 0 && i <ndim():索引= 0必须在[0,-1)范围内

时间:2020-02-18 20:01:23

标签: python mxnet

我创建了训练有素的模型,现在正在尝试进行预测。我加载了模型和预测数据。 我的预测文件是包含3列的CSV文件。这是我的代码

mod = mx.module.Module.load("mx-mod", 0)
data_iter = mx.io.CSVIter(data_csv=os.path.join(ML_DIR_LOCAL, "predict.csv"),
                          batch_size=1,
                          data_shape=(3, ))
mod.bind(data_shapes=data_iter.provide_data)
mod.predict(data_iter)

bind方法失败,并显示以下错误: Check failed: i >= 0 && i < ndim(): index = 0 must be in range [0, -1) 我不明白

0 个答案:

没有答案