添加AUC作为keras的损失函数

时间:2015-09-11 08:46:44

标签: theano keras

有没有人为使用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 个答案:

答案 0 :(得分:6)

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