Keras添加auc metrics

手工写一个类auc继承于Callback类
里面的函数应该是override,名称都很一目了然
on_train_begin
on_train_end
on_epoch_begin
on_epoch_end
我们的逻辑就写在on_epoch_end下,计算AUC,并格式化输出
具体的面向对象机制,我可能需要找个时间学习下
在model.fit中callbacks参数(是一个list)传入实例auc实例

留言

熱門文章