imblearn 分类报告中的 IBA 是什么意思?

数据挖掘 Python 分类 阶级失衡 不平衡学习
2022-01-23 18:23:43

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 代表几何平均值。所有这些都是衡量不平衡类性能的指标。

1个回答

如果您查看classification_report_imbalancedimblearn的文档,您可以看到它代表“索引平衡精度”。有关索引平衡准确度是什么以及它在不平衡数据集上的价值的更多信息,请查看原始论文iba