我想预测双分类问题的概率。以前我使用model.predict_proba或predict_on_batch来解决这个问题。现在我想在我的脚本中使用生成器,但是我找不到像evaluate_generator或predict_generator这样的生成器。 evaluate_generator或predict_generator都不会生成概率。什么是Keras中概率预测的生成方法?
答案 0 :(得分:0)
无论输出是否是概率取决于实际的神经网络模型,而不是predict_generator
。如果您的模型已经输出概率,意味着它在输出处有一个softmax激活,那么使用predict_generator
应该给出概率值。