有没有人有幸为使用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 回答
AUC不是可微分的,所以你可以在算法上最大化AUC,但我建议只使用常规的交叉熵/对数似然丢失 .