imblearn 是一个用于处理不平衡数据的 python 库。下面给出了生成分类报告的代码。
import numpy as np
from imblearn.metrics import classification_report_imbalanced
y_true = [0, 1, 2, 2, 2]
y_pred = [0, 0, 2, 2, 1]
target_names = ['class 0', 'class 1', 'class 2']
print(classification_report_imbalanced(y_true, y_pred,target_names=target_names))
输出如下
pre rec spe f1 geo iba sup
class 0 0.50 1.00 0.75 0.67 0.87 0.77 1
class 1 0.00 0.00 0.75 0.00 0.00 0.00 1
class 2 1.00 0.67 1.00 0.80 0.82 0.64 3
avg/total 0.70 0.60 0.90 0.61 0.66 0.54 5
这个分类报告中的iba是什么意思。这里 pre 代表精确度,rec 代表召回率,spe 代表特异性,f1 代表 f1 度量,geo 代表几何平均值。所有这些都是衡量不平衡类性能的指标。