标签： python classification prediction mnist lasagne
我正在尝试对mnist数据集进行分类，它的工作非常好。为了使用预训练模型对图像进行分类，我看了一下这个例子：
Theano/lasagne: how to predict with trained model
使用命令
result = T.argmax（预测，轴= 1）
predict_fn = theano.function（[input_var]，result）
我得到了正确的分类，但我只得到了结果。有人可以帮助我，我怎么能得到这门课的概率？
感谢您的帮助！