使用mxnet CNN模型进行预测

Question

嗨，我是数据科学的新手， 我遵循了本教程https://mxnet.incubator.apache.org/tutorials/nlp/cnn.html，但我对如何使用上述教程生成的训练模型进行单一预测感到困惑。请指导我正确的方向来解决这个问题。感谢。

Answer 1

以下是如何使用预训练模型进行预测的教程：https://mxnet.incubator.apache.org/tutorials/python/predict_image.html

步骤：
1.加载预训练模型并创建MXNet模块实例 2.抓取您的数据并继续使用模块

Answer 2

可以像这样加载模型：

sym, arg_params, aux_params = mx.model.load_checkpoint('cnn', 3)
mod = mx.mod.Module(symbol=sym, context=mx.cpu(), label_names=None)
mod.bind(for_training=False, data_shapes=[('data', (50,56))], 
         label_shapes=mod._label_shapes)
mod.set_params(arg_params, aux_params, allow_missing=True)

但不幸的是，你不能用它来做一个预测。你需要一批50个。