Keras 自定义Callback

参考:回调函数Callback - Keras中文文档

keras 上添加 roc auc指標 - 掃文資訊

要在keras中定义一个计算roc_auc得分的回掉函数

先自定义Callback类


Keras 自定义Callback_第1张图片

实例化并且放到 fit 函数中

Keras 自定义Callback_第2张图片

这种方法有个缺点就是,需要额外对训练集多做一次预测来计算auc得分

你可能感兴趣的:(Keras 自定义Callback)