小编blo*_*oop的帖子

roc_auc_score - y_true中只有一个类

我在现有的数据帧上做了k-fold XV,我需要获得AUC分数.问题是 - 有时测试数据只包含0,而不是1!

我尝试使用示例,但使用不同的数字:

import numpy as np
from sklearn.metrics import roc_auc_score
y_true = np.array([0, 0, 0, 0])
y_scores = np.array([1, 0, 0, 0])
roc_auc_score(y_true, y_scores)
Run Code Online (Sandbox Code Playgroud)

我得到这个例外:

ValueError:y_true中只有一个类.在这种情况下,没有定义ROC AUC分数.

是否有任何解决方法可以使其在这种情况下工作?

python scikit-learn auc

8
推荐指数
1
解决办法
9953
查看次数

标签 统计

auc ×1

python ×1

scikit-learn ×1