jer*_*_K7 3 python precision machine-learning scikit-learn precision-recall
我正在研究一个多分类项目,我注意到无论我运行什么分类器,模型中的精度和召回率都是相同的。
分类问题具有三个不同的类别。数据量偏小,有 13k 个实例,分为测试 (0.8) 和训练 (0.2)。
训练数据的形状为 (10608, 28),标签的形状为 (10608, 3)(二值化标签)。
分类不平衡:
我正在比较不同的分类器,以便稍后关注最有前途的分类器。在计算每个模型的精度和召回率时,我注意到它们在模型中总是相同的。
由于精确率和召回率的计算方式不同,当假阴性预测的数量等于假阳性预测的数量时,它们可能是相同的 FP = FN。
例子:
sgd_clf = OneVsRestClassifier(SGDClassifier(random_state=42))
sgd_clf.fit(data_tr, labels_tr)
y_pred_sgd = cross_val_predict(sgd_clf, data_tr, labels_tr, cv=5)
cm_sgd = confusion_matrix(labels_tr.argmax(axis=1), y_pred_sgd.argmax(axis=1))
cm_sgd:
array([[1038, 19, 2084],
[ 204, 22, 249],
[ 931, 48, 6013]], dtype=int64)
precision_score(labels_tr.argmax(axis=1), y_pred_sgd.argmax(axis=1), average="micro")
0.666760935143288
recall_score(labels_tr.argmax(axis=1), y_pred_sgd.argmax(axis=1), average="micro")
0.666760935143288
FP=FN=3535
Run Code Online (Sandbox Code Playgroud)
lr_clf = OneVsRestClassifier(LogisticRegression(random_state=42, max_iter=4000))
lr_clf.fit(data_tr, labels_tr)
y_pred_lr = cross_val_predict(lr_clf, data_tr, labels_tr, cv=5)
cm_lr = confusion_matrix(labels_tr.argmax(axis=1), y_pred_lr.argmax(axis=1))
cm_lr:
array([[ 982, 1, 2158],
[ 194, 7, 274],
[ 774, 9, 6209]], dtype=int64)
precision_score(labels_tr.argmax(axis=1), y_pred_lr.argmax(axis=1), average="micro")
0.6785444947209653
recall_score(labels_tr.argmax(axis=1), y_pred_lr.argmax(axis=1), average="micro")
0.6785444947209653
FP=FN=3410
Run Code Online (Sandbox Code Playgroud)
rf_clf = OneVsRestClassifier(RandomForestClassifier(random_state=42))
rf_clf.fit(data_tr, labels_tr)
y_pred_forest = cross_val_predict(rf_clf, data_tr, labels_tr, cv=5)
cm_forest = confusion_matrix(labels_tr.argmax(axis=1), y_pred_forest.argmax(axis=1))
cm_forest:
array([[1576, 56, 1509],
[ 237, 45, 193],
[1282, 61, 5649]], dtype=int64)
precision_score(labels_tr.argmax(axis=1), y_pred_forest.argmax(axis=1), average="micro")
0.6853318250377074
recall_score(labels_tr.argmax(axis=1), y_pred_forest.argmax(axis=1), average="micro")
0.6853318250377074
FP=FN=3338
Run Code Online (Sandbox Code Playgroud)
模型中所有模型具有相同召回率和精度的可能性有多大?我错过了什么吗?
发生这种情况是因为您正在计算micro分数的平均值。在文档中,它被描述为:
通过计算总的真阳性、假阴性和假阳性来计算全局指标。
现在有一个问题:在分类任务中,每个测试用例都保证被分配到一个类别,计算平均值micro相当于计算准确性分数。这就是为什么您在每个模型中获得相同的精度和召回结果:您基本上是在计算所有情况下的准确性。
您可以通过使用accuracy_score和比较结果来验证这一点。
因此,您应该使用其中之一macro或weighted平均值来更好地评估模型的精确度和召回率。
| 归档时间: |
|
| 查看次数: |
2179 次 |
| 最近记录: |