【发布时间】:2018-02-09 06:05:34
【问题描述】:
我正在使用逻辑回归方法进行多类分类。基本上我知道,如果我使用 accuracy_score () 函数(例如,来自 sklearn 库),它会计算出不同值到不同值的准确度,如下所示:
y_pred = [0, 2, 1, 3]
y_true = [0, 1, 2, 3]
accuracy_score(y_true, y_pred)
0.5
但我想获取每个标签(来自 predict_proba)的顶级逻辑回归预测向量的 accuracy_score_new () 函数,并计算真实标签是否在此区间内,如下所示:
y_pred = [[0,1,3] [2,1,4], [1,2,5] [3,7,9]]
y_true = [0, 1, 2, 3]
accuracy_score_new(y_true, y_pred)
1
此示例中的 accuracy_score_new 将等于 1,因为分类器预测标签在区间内。这个功能怎么实现?
【问题讨论】:
-
您是想在任何 scikit 类中使用新函数,还是只想自己使用它来打印新的精度。如果是后者,那么您可以为它编写简单的 python 代码。只需遍历两个数组。
标签: python scikit-learn classification logistic-regression multiclass-classification