ロジスティック回帰を使って確率を予測したいときに「評価メトリクスとして使いたいのは AUC (areaUnderROC) じゃなくて Logarithmic loss (LogLoss) なんだよ!」と常々思っているのですが、現在の MLlib には二値分類 (BinaryClassificationEvaluator
) 、多クラス分類 (MulticlassClassificationEvaluator
)、回帰 (RegressionEvaluator
) 用の Evaluator
実装しか用意されていなかったので、ついカッとなって実装してしまった次第です。
これくらいの機能は標準で取り揃えていてもいいんじゃないかと思うんだけどなあ…
0 コメント:
コメントを投稿