Keras中的自定义预测方法

时间:2017-04-13 12:06:29

标签: python deep-learning keras

我想在给定的Kears模型中使用自定义预测方法。

例如,假设我有一个二元分类器,我的预测规则说如果一个例子属于A类的概率小于0.8,那么它应该被标记为B类(而不是当前的0.5)。

有没有办法在Keras中覆盖predict方法而不使用predict_proba外部方法然后应用自定义预测规则?

0 个答案:

没有答案