2017-08-30 601 views
0

我想预测双分类问题的概率。以前我使用model.predict_proba或predict_on_batch来解决这个问题。现在我想在脚本中使用生成器,但是我找不到像evaluate_generator或predict_generator这样的生成器。 evaluate_generator或predict_generator都不会生成概率。 Keras中用于概率预测的生成器方法是什么?Keras中概率预测的生成方法是什么?

+0

您使用哪种Keras版本?看起来Keras具有'evaluate_generator'和'predict_generator',并且'predict_generator'返回Numpy数组的预测(概率)。参见[Keras Sequential](https://keras.io/models/sequential/) –

回答

0

无论输出是否为概率取决于实际的神经网络模型,而不取决于predict_generator。如果您的模型已经输出了概率,这意味着它在输出端具有softmax激活,那么使用predict_generator应该给出概率值。