6

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

所以我想我的问题是,有没有人能够编写这个函数?你愿意分享吗?

4

1 回答 1

7

AUC 是不可微的,所以如果不做一些修改,就不能将其用作损失函数。有一些关于最大化 AUC 的算法的工作,但我建议只使用常规的交叉熵/对数似然损失。

于 2016-05-01T19:10:03.990 回答