添加AUC作为keras的损失函数

Question

有没有人有幸为使用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

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

Answer 1

AUC不可区分,因此如果不进行一些修改,您不能将其用作损失函数.已经有一些算法可以最大化AUC,但我建议只使用常规的交叉熵/对数似然丢失.