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 に答える