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.
2 Answers
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
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.
Related