什么是Keras中概率预测的生成方法?

时间:2017-08-30 04:58:37

标签: keras generator probability prediction

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

1 个答案:

答案 0 :(得分:0)

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