首页 > 解决方案 > 如何获得适用于 sklearn 的日志损失指标的多标签(不是多类)分类?

问题描述

我想我只是不理解 sklearn.metrics.log_loss 的 API

演示该问题的小代码示例:

import numpy as np
import sklearn.metrics
foo = np.asarray([[0, 0], [0, 1]])

print(sklearn.metrics.log_loss(foo, np.eye(2)))
print(sklearn.metrics.log_loss(foo, foo))

输出:

4.996003610813207e-16
4.996003610813207e-16

这是没有意义的。y_pred 不匹配。

旁注,我知道 y_pred 会被剪掉,但这并没有真正改变我的问题。对损失的截断贡献是:

-np.log(1e-15)
34.538776394910684

标签: pythonscikit-learnloss-functionmultilabel-classification

解决方案


推荐阅读