使用mxnet CNN模型进行预测

Question

Hi I'm a newbie to data science, I followed this tutorial https://mxnet.incubator.apache.org/tutorials/nlp/cnn.html but I am confused over how to make a single prediction using the trained model generated by the above mentioned tutorial. Please guide me the right direction to fix this. Thanks.

Answer 1

Here is a tutorial of how to predict with pretrained model: https://mxnet.incubator.apache.org/tutorials/python/predict_image.html

Steps:
1. Load pretrained model and create a MXNet module instance.
2. Grab your data and run forward with module

Answer 2

One can load the model like this:

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)

But unfortunately, you cannot use this for making only one prediction. You would need a batch of 50.