小编use*_*452的帖子

我试图做一个简单的概念验证,在那里我可以看到给定预测的不同类的概率.

但是,我尝试的所有内容似乎只输出预测的类,即使我使用softmax激活.我是机器学习的新手,所以我不确定我是否犯了一个简单的错误,或者这是否是Keras中没有的功能.

我正在使用Keras + TensorFlow.我已经调整了Keras给出的一个基本示例,用于对MNIST数据集进行分类.

下面的代码与示例完全相同,只是一些(注释的)额外行将模型导出到本地文件.

'''Trains a simple deep NN on the MNIST dataset.
Gets to 98.40% test accuracy after 20 epochs
(there is *a lot* of margin for parameter tuning).
2 seconds per epoch on a K520 GPU.
'''

from __future__ import print_function

import keras
from keras.datasets import mnist
from keras.models import Sequential
from keras.layers import Dense, Dropout
from keras.optimizers import RMSprop

import h5py # added import because it is required for model.save
model_filepath = 'test_model.h5' …