我创建了训练有素的模型,现在正在尝试进行预测。我加载了模型和预测数据。 我的预测文件是包含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)
我不明白