处理不平衡数据的十大 Python 库
共 3139字，需浏览 7分钟
·
2024-04-18 17:20
↓推荐关注↓
1、imbalanced-learn
imbalanced-learn是scikit-learn的扩展，提供了各种重新平衡数据集的技术。它提供过采样、欠采样和组合方法。
from imblearn.over_sampling import RandomOverSampler
ros = RandomOverSampler()
X_resampled, y_resampled = ros.fit_resample(X, y)
2、SMOTE
SMOTE生成合成样本来平衡数据集。
from imblearn.over_sampling import SMOTE
smote = SMOTE()
X_resampled, y_resampled = smote.fit_resample(X, y)
3、ADASYN
ADASYN根据少数样本的密度自适应生成合成样本。
from imblearn.over_sampling import ADASYN
adasyn = ADASYN()
X_resampled, y_resampled = adasyn.fit_resample(X, y)
4、RandomUnderSampler
RandomUnderSampler随机从多数类中移除样本。
from imblearn.under_sampling import RandomUnderSampler
rus = RandomUnderSampler()
X_resampled, y_resampled = rus.fit_resample(X, y)
5、Tomek Links
Tomek Links可以移除的不同类的最近邻居对，减少多样本的数量
from imblearn.under_sampling import TomekLinks
tl = TomekLinks()
X_resampled, y_resampled = tl.fit_resample(X, y)
6、SMOTEENN (SMOTE +Edited Nearest Neighbors)
SMOTEENN结合SMOTE和Edited Nearest Neighbors。
from imblearn.combine import SMOTEENN
smoteenn = SMOTEENN()
X_resampled, y_resampled = smoteenn.fit_resample(X, y)
7、SMOTETomek (SMOTE + Tomek Links)
SMOTEENN结合SMOTE和Tomek Links进行过采样和欠采样。
from imblearn.combine import SMOTETomek
smotetomek = SMOTETomek()
X_resampled, y_resampled = smotetomek.fit_resample(X, y)
8、EasyEnsemble
EasyEnsemble是一种集成方法，可以创建多数类的平衡子集。
from imblearn.ensemble import EasyEnsembleClassifier
ee = EasyEnsembleClassifier()
ee.fit(X, y)
9、BalancedRandomForestClassifier
BalancedRandomForestClassifier是一种将随机森林与平衡子样本相结合的集成方法。
from imblearn.ensemble import BalancedRandomForestClassifier
brf = BalancedRandomForestClassifier()
brf.fit(X, y)
10、RUSBoostClassifier
RUSBoostClassifier是一种结合随机欠采样和增强的集成方法。
from imblearn.ensemble import RUSBoostClassifier
rusboost = RUSBoostClassifier()
rusboost.fit(X, y)
总结
处理不平衡数据对于建立准确的机器学习模型至关重要。这些Python库提供了各种技术来应对这一问题。根据你的数据集和问题，可以选择最合适的方法来有效地平衡数据。
长按或扫描下方二维码，后台回复：加群，即可申请入群。一定要备注：来源+研究方向+学校/公司，否则不拉入群中，见谅！
（长按三秒，进入后台）
推荐阅读
评论