imbalanced-learn，一个强大的 Python 库！

大家好，我是涛哥，本文内容来自 涛哥聊Python ，转载请标原创。

今天为大家分享一个强大的 Python 库 - imbalanced-learn

Github地址：https://github.com/scikit-learn-contrib/imbalanced-learn

---

在实际的数据分析和机器学习任务中，经常会遇到数据不平衡的情况，即不同类别的样本数量差异较大，这会导致模型训练和预测的不准确性。Python的imbalanced-learn库提供了一系列处理不平衡数据的方法和工具，帮助开发者更好地应对这一问题。本文将深入探讨imbalanced-learn库的特性、安装方法、基本功能、高级功能、实际应用场景以及总结，带领大家全面了解这个解决数据不平衡的利器。

安装

安装imbalanced-learn库可以使用pip命令：

pip install imbalanced-learn

安装完成后，就可以开始使用imbalanced-learn库了。

特性

• 提供了多种处理不平衡数据的方法，如过采样、欠采样、合成数据等。
• 兼容scikit-learn库，可以与scikit-learn的模型无缝集成。
• 支持多种评估指标，如准确率、召回率、F1分数等。

基本功能

1. 过采样

过采样是解决数据不平衡的常用方法之一，imbalanced-learn库提供了多种过采样算法，如SMOTE、ADASYN等，示例代码如下：

from imblearn.over_sampling import SMOTE
from sklearn.datasets import make_classification
from collections import Counter

# 创建不平衡数据集
X, y = make_classification(n_classes=2, class_sep=2, weights=[0.1, 0.9], n_informative=3, n_redundant=1, flip_y=0, n_features=20, n_clusters_per_class=1, n_samples=1000, random_state=10)

# 打印原始数据集样本分布
print("原始数据集样本分布:", Counter(y))

# 使用SMOTE算法进行过采样
smote = SMOTE(random_state=42)
X_resampled, y_resampled = smote.fit_resample(X, y)

# 打印过采样后数据集样本分布
print("过采样后数据集样本分布:", Counter(y_resampled))

2. 欠采样

除了过采样，imbalanced-learn库还支持欠采样方法，如RandomUnderSampler、NearMiss等，示例代码如下：

from imblearn.under_sampling import RandomUnderSampler
from sklearn.datasets import make_classification
from collections import Counter

# 创建不平衡数据集
X, y = make_classification(n_classes=2, class_sep=2, weights=[0.1, 0.9], n_informative=3, n_redundant=1, flip_y=0, n_features=20, n_clusters_per_class=1, n_samples=1000, random_state=10)

# 打印原始数据集样本分布
print("原始数据集样本分布:", Counter(y))

# 使用RandomUnderSampler算法进行欠采样
rus = RandomUnderSampler(random_state=42)
X_resampled, y_resampled = rus.fit_resample(X, y)

# 打印欠采样后数据集样本分布
print("欠采样后数据集样本分布:", Counter(y_resampled))

高级功能

imbalanced-learn库还提供了一些高级功能，如合成数据、集成学习等，可以进一步提升模型的性能和鲁棒性。

1. 合成数据

imbalanced-learn库中的合成数据方法可以根据原始数据集生成新的合成数据，从而增加样本多样性，示例代码如下：

from imblearn.datasets import make_imbalance
from sklearn.datasets import make_classification
from collections import Counter

# 创建不平衡数据集
X, y = make_classification(n_classes=2, class_sep=2, weights=[0.1, 0.9], n_informative=3, n_redundant=1, flip_y=0, n_features=20, n_clusters_per_class=1, n_samples=1000, random_state=10)

# 打印原始数据集样本分布
print("原始数据集样本分布:", Counter(y))

# 使用make_imbalance方法生成合成数据
X_resampled, y_resampled = make_imbalance(X, y, sampling_strategy={0: 500, 1: 500})

# 打印合成数据集样本分布
print("合成数据集样本分布:", Counter(y_resampled))

2. 集成学习

imbalanced-learn库中的集成学习方法可以结合多个模型进行集成，提高模型的泛化能力和稳定性，示例代码如下：

from imblearn.ensemble import BalancedRandomForestClassifier
from sklearn.datasets import make_classification
from sklearn.model_selection import train_test_split
from sklearn.metrics import classification_report

# 创建不平衡数据集
X, y = make_classification(n_classes=2, class_sep=2, weights=[0.1, 0.9], n_informative=3, n_redundant=1, flip_y=0, n_features=20, n_clusters_per_class=1, n_samples=1000, random_state=10)

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

# 使用BalancedRandomForestClassifier进行集成学习
clf = BalancedRandomForestClassifier(n_estimators=100, random_state=42)
clf.fit(X_train, y_train)
y_pred = clf.predict(X_test)

# 打印分类报告
print(classification_report(y_test, y_pred))

实际应用场景

1. 金融欺诈检测

在金融领域，欺诈检测是一个重要的问题。由于正常交易与欺诈交易的比例通常是不平衡的，因此可以使用imbalanced-learn库中的过采样或欠采样方法来处理数据不平衡问题，进而训练出更准确的欺诈检测模型。

示例代码：

from imblearn.over_sampling import SMOTE
from sklearn.datasets import make_classification
from sklearn.model_selection import train_test_split
from sklearn.ensemble import RandomForestClassifier
from sklearn.metrics import accuracy_score, precision_score, recall_score, f1_score

# 创建不平衡数据集
X, y = make_classification(n_classes=2, class_sep=2, weights=[0.1, 0.9], n_informative=3, n_redundant=1, flip_y=0, n_features=20, n_clusters_per_class=1, n_samples=1000, random_state=10)

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

# 使用SMOTE进行过采样
smote = SMOTE(random_state=42)
X_resampled, y_resampled = smote.fit_resample(X_train, y_train)

# 训练随机森林分类器
clf = RandomForestClassifier(random_state=42)
clf.fit(X_resampled, y_resampled)
y_pred = clf.predict(X_test)

# 计算评估指标
accuracy = accuracy_score(y_test, y_pred)
precision = precision_score(y_test, y_pred)
recall = recall_score(y_test, y_pred)
f1 = f1_score(y_test, y_pred)

# 打印评估指标
print("准确率:", accuracy)
print("精确率:", precision)
print("召回率:", recall)
print("F1分数:", f1)

2. 医疗诊断

在医疗领域，数据不平衡也是常见的情况。例如，在医疗诊断中，某种疾病的患者与非患者的比例可能是不平衡的。可以利用imbalanced-learn库中的方法来处理这种数据不平衡，训练出更准确的医疗诊断模型。

示例代码：

from imblearn.under_sampling import RandomUnderSampler
from sklearn.datasets import make_classification
from sklearn.model_selection import train_test_split
from sklearn.svm import SVC
from sklearn.metrics import accuracy_score, precision_score, recall_score, f1_score

# 创建不平衡数据集
X, y = make_classification(n_classes=2, class_sep=2, weights=[0.1, 0.9], n_informative=3, n_redundant=1, flip_y=0, n_features=20, n_clusters_per_class=1, n_samples=1000, random_state=10)

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

# 使用RandomUnderSampler进行欠采样
rus = RandomUnderSampler(random_state=42)
X_resampled, y_resampled = rus.fit_resample(X_train, y_train)

# 训练支持向量机分类器
clf = SVC(random_state=42)
clf.fit(X_resampled, y_resampled)
y_pred = clf.predict(X_test)

# 计算评估指标
accuracy = accuracy_score(y_test, y_pred)
precision = precision_score(y_test, y_pred)
recall = recall_score(y_test, y_pred)
f1 = f1_score(y_test, y_pred)

# 打印评估指标
print("准确率:", accuracy)
print("精确率:", precision)
print("召回率:", recall)
print("F1分数:", f1)

3. 文本分类

在文本分类任务中，不同类别的文本样本数量可能会存在不平衡。可以利用imbalanced-learn库中的方法处理这种数据不平衡情况，训练出更准确的文本分类模型。

示例代码：

from imblearn.combine import SMOTETomek
from sklearn.datasets import fetch_20newsgroups
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.model_selection import train_test_split
from sklearn.naive_bayes import MultinomialNB
from sklearn.metrics import accuracy_score, precision_score, recall_score, f1_score

# 加载数据集
newsgroups = fetch_20newsgroups(subset='all')

# 文本特征提取
vectorizer = TfidfVectorizer()
X = vectorizer.fit_transform(newsgroups.data)
y = newsgroups.target

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

# 使用SMOTETomek进行合成采样
smote_tomek = SMOTETomek(random_state=42)
X_resampled, y_resampled = smote_tomek.fit_resample(X_train, y_train)

# 训练朴素贝叶斯分类器
clf = MultinomialNB()
clf.fit(X_resampled, y_resampled)
y_pred = clf.predict(X_test)

# 计算评估指标
accuracy = accuracy_score(y_test, y_pred)
precision = precision_score(y_test, y_pred, average='weighted')
recall = recall_score(y_test, y_pred, average='weighted')
f1 = f1_score(y_test, y_pred, average='weighted')

# 打印评估指标
print("准确率:", accuracy)
print("精确率:", precision)
print("召回率:", recall)
print("F1分数:", f1)

总结

通过本文对imbalanced-learn库的介绍，了解了该库的安装方法、基本功能、高级功能以及实际应用场景。imbalanced-learn库为处理数据不平衡提供了丰富的工具和方法，能够有效提升模型的性能和稳定性，适用于多种领域的数据处理任务。在实际应用中，开发者可以根据具体情况选择合适的方法和技术，从而更好地应对数据不平衡的挑战。