首页 文章

添加AUC作为keras的损失函数

提问于
浏览
5

有没有人有幸为使用Theano为Keras编写自定义AUC丢失功能?

文档在这里:http://keras.io/objectives/

示例代码在这里:https://github.com/fchollet/keras/blob/master/keras/objectives.py

我看到pylearn2中有一个实现,(它实际上是sklearn的一个包装器)但是无法将其移植到Keras中使用

https://github.com/lisa-lab/pylearn2/blob/master/pylearn2/train_extensions/roc_auc.py

所以我想我的问题是,是否有人能够写这个功能?你会愿意分享吗?

1 回答

  • 6

    AUC不是可微分的,所以你可以在算法上最大化AUC,但我建议只使用常规的交叉熵/对数似然丢失 .

相关问题