机器学习-分类不平衡数据集-过采样-欠采样

一、创建分类不平衡的数据集及进行可视化:

1、所需API:

from sklearn.datasets import make_classification
import matplotlib.pyplot as plt
from collections import Counter

2、代码演示🌰:使用make_classification生成样本数据
from sklearn.datasets import make_classification
import matplotlib.pyplot as plt
from collections import Counter

# 1.创建类别不平衡的数据集:
# 使用make_classification生成样本数据
X, y = make_classification(n_samples=5000,
                           n_features=2,  # 特征个数 = n_informative() + n_redundant + n_repeated
                           n_informative=2,  # 多信息特征的个数
                           n_redundant=0,  # 冗余信息,informative特征的随机线性组合
                           n_repeated=0,  # 重复信息,随机提取n_informative和n_redundant 特征
                           n_classes=3,  # 分类类别
                           n_clusters_per_class=1,  # 某一个类别是有几个cluster构成的
                           weights=[0.01, 0.05, 0.94],  # 列表类型,权重比
                           random_state=0
                           )

# 2.查看各个标签的样本:
counter = Counter(y)
print(counter) # Counter({2: 4674, 1: 262, 0: 64})

# 3.数据集可视化:
plt.scatter(X[:, 0], X[:, 1], c=y)
plt.show()

3、运行结果:
运行结果:.png

二、过采样和欠采样

1、过采样方法:

对训练集里的少数类进行“过采样”(oversampling),既增加一些少数类样本🙆正、反例数目接近,然后再进行学习。

1.1、随机过采样方法:
  • 方法:随机过采样是在少数类Smin中随机选择一些样本,然后通过复制所选择的样本生成样本集E,将他们添加到Smin中扩大原始数据集从而得到新的少数类集合Snew-min。新数据集Snew-min=Smin + E

  • API:from imblearn.over_sampling import RandomOverSampler

  • 代码演示🌰:

from sklearn.datasets import make_classification
import matplotlib.pyplot as plt
from collections import Counter
from imblearn.over_sampling import RandomOverSampler

# 1.创建类别不平衡的数据集:
# 使用make_classification生成样本数据
X, y = make_classification(n_samples=5000,
                           n_features=2,  # 特征个数 = n_informative() + n_redundant + n_repeated
                           n_informative=2,  # 多信息特征的个数
                           n_redundant=0,  # 冗余信息,informative特征的随机线性组合
                           n_repeated=0,  # 重复信息,随机提取n_informative和n_redundant 特征
                           n_classes=3,  # 分类类别
                           n_clusters_per_class=1,  # 某一个类别是有几个cluster构成的
                           weights=[0.01, 0.05, 0.94],  # 列表类型,权重比
                           random_state=0
                           )

# 2.查看各个标签的样本:
counter = Counter(y)
print(counter)  # Counter({2: 4674, 1: 262, 0: 64})

# 3.数据集可视化:
plt.scatter(X[:, 0], X[:, 1], c=y)
plt.show()

# 4、过采样 
# 4.1 随机过采样
ros = RandomOverSampler(random_state=0)

X_resampled, y_resampled = ros.fit_resample(X, y)
counter_resampled = Counter(y_resampled)
print(counter_resampled)

plt.scatter(X_resampled[:, 0], X_resampled[:, 1], c=y_resampled)
plt.show()

  • 随机过采样运行结果:
随机过采样运行结果.png
  • 随机过采样的缺点:
    (1)对于随机过采样,由于需要对少数类样本进行复制来扩大数据集,造成模型训练复杂度加大。
    (2)另一方面也容易造成模型的过拟合问题,因为随机过采样是简单的对初始样本进行复制采样,这就使得学习器学得的规则过于具体化,不利于学习器的泛化性能,造成过拟合问题。
    (3)为了解决随机过采样中造成模型过拟合问题,又能保证实现数据集均衡的目的,出现了过采样法代表性的算法SMOTE算法
1.2、SMOTE算法(过采样代表性算法):

1.2.1、SMOTE全称是Synthetic Minority Oversampling即合成少数类过采样技术。是对随机过采样方法的一个改进算法,由于随机过采样方法是直接对少数类进行重采用,会使训练集中有很多重复的样本,容易造成产生的模型过拟合问题。

而SMOTE算法的基本思想:对每个少数类样本 xi ,从它的最近邻中随机选择一个样本 x̅i (即x拔), x̅i 是少数类中的一个样本,然后在 xix̅i 之间的连线上随机选择一个点作为新合成的少数样本类。

1.2.2、SMOTE算法合成新少数类样本的算法过程:
(1)对少数类中的每一个样本xi,以欧式距离为标准计算它到少数类样本集Smin中所有样本的距离,得到其k近邻。
(2)根据样本不平衡比例设置一个采样比例以确定采样倍率N,对于每一个少数类样本 xi ,从其k近邻中随机选择若干个样本,假设选择的是 x̅i
(3)对于每一个随机选择出来的近邻 x̅i ,分别与 xi 按照如下公式构建新的样本。

Xnew = Xi + rand(0, 1) × (x̅i - Xi)

1.2.3、图文表达方式:

(1)先随机选定一个少数类样本 Xi

1.png

(2)找出这个少数类样本 Xi 的K个近邻(假设K=5),5个近邻已经被圈出。

2.png

(3)随机从这K个近邻中选择一个样本 x̅i (用绿色圈出来了)。

3.png

(4)在少数类样本 Xi 和被选中的这个近邻样本 x̅i 之间的连线上,随机找一点。这个点就是人工合成的新的样本点(绿色正号标出)。

4.png
SMOTE算法摒弃了随机过采样复制样本的做法,可以防止随机过采样中容易过拟合的问题,实践证明此方法可以提高分类器的性能。

1.2.4、SMOTE算法API:from imblearn.over_sampling import SMOTE

1.2.5、SMOTE算法代码实现🌰:

from sklearn.datasets import make_classification
import matplotlib.pyplot as plt
from collections import Counter
from imblearn.over_sampling import RandomOverSampler, SMOTE
from imblearn.under_sampling import RandomUnderSampler

# 1.创建类别不平衡的数据集:
# 使用make_classification生成样本数据
X, y = make_classification(n_samples=5000,
                           n_features=2,  # 特征个数 = n_informative() + n_redundant + n_repeated
                           n_informative=2,  # 多信息特征的个数
                           n_redundant=0,  # 冗余信息,informative特征的随机线性组合
                           n_repeated=0,  # 重复信息,随机提取n_informative和n_redundant 特征
                           n_classes=3,  # 分类类别
                           n_clusters_per_class=1,  # 某一个类别是有几个cluster构成的
                           weights=[0.01, 0.05, 0.94],  # 列表类型,权重比
                           random_state=0
                           )

# 2.查看各个标签的样本:
counter = Counter(y)
print(counter)  # Counter({2: 4674, 1: 262, 0: 64})

# 3.数据集可视化:
plt.scatter(X[:, 0], X[:, 1], c=y)
plt.show()

# 4、过采样
# 4.1 随机过采样
ros = RandomOverSampler(random_state=0)

X_resampled, y_resampled = ros.fit_resample(X, y)
counter_resampled = Counter(y_resampled)
print(counter_resampled)

plt.scatter(X_resampled[:, 0], X_resampled[:, 1], c=y_resampled)
plt.show()

# 4.2 SMOTE过采样
X_resampled2, y_resampled2 = SMOTE().fit_resample(X, y)
counter_resampled2 = Counter(y_resampled2)
print("SMOTE过采样结果:\n", counter_resampled2)

plt.scatter(X_resampled2[:, 0], X_resampled2[:, 1], c=y_resampled2)
plt.show()

1.2.6、SMOTE过采样运行结果:

SMOTE过采样运行结果.png
2、欠采样方法:

2.1、什么是欠采样方法:

直接对训练集中多数类样本进行“欠采样”(undersampling),即去除一些多数类中的样本使得正例、反例数目接近,然后再进行学习。

2.2、随机欠采样方法:

随机欠采样即从多数类 Smaj 中随机选择一些样本组成样本集 E 。然后将样本集 ESmaj 中移除。新的数据集 Snew-maj = Smaj - E

2.3、随机欠采样方法缺点:
随机欠采样方法通过改变多数类样本比例以达到修改样本分布的目的,从而使样本分布较为均衡,但是这也存在一些问题。对于随机欠采样,由于采样的样本集合要少于原来的样本集合,因此会造成一些信息缺失,即将多数类样本删除有可能会导致分类器丢失有关多数类的重要信息。

2.4、随机欠采样方法的代码实现🌰:

from sklearn.datasets import make_classification
import matplotlib.pyplot as plt
from collections import Counter
from imblearn.over_sampling import RandomOverSampler, SMOTE
from imblearn.under_sampling import RandomUnderSampler

# 1.创建类别不平衡的数据集:
# 使用make_classification生成样本数据
X, y = make_classification(n_samples=5000,
                           n_features=2,  # 特征个数 = n_informative() + n_redundant + n_repeated
                           n_informative=2,  # 多信息特征的个数
                           n_redundant=0,  # 冗余信息,informative特征的随机线性组合
                           n_repeated=0,  # 重复信息,随机提取n_informative和n_redundant 特征
                           n_classes=3,  # 分类类别
                           n_clusters_per_class=1,  # 某一个类别是有几个cluster构成的
                           weights=[0.01, 0.05, 0.94],  # 列表类型,权重比
                           random_state=0
                           )

# 2.查看各个标签的样本:
counter = Counter(y)
print(counter)  # Counter({2: 4674, 1: 262, 0: 64})

# 3.数据集可视化:
plt.scatter(X[:, 0], X[:, 1], c=y)
plt.show()

# 4、过采样
# 4.1 随机过采样
ros = RandomOverSampler(random_state=0)

X_resampled, y_resampled = ros.fit_resample(X, y)
counter_resampled = Counter(y_resampled)
print(counter_resampled)

plt.scatter(X_resampled[:, 0], X_resampled[:, 1], c=y_resampled)
plt.show()

# 4.2 SMOTE过采样
X_resampled2, y_resampled2 = SMOTE().fit_resample(X, y)
counter_resampled2 = Counter(y_resampled2)
print("SMOTE过采样结果:\n", counter_resampled2)

plt.scatter(X_resampled2[:, 0], X_resampled2[:, 1], c=y_resampled2)
plt.show()

# 5、欠采样
# 5.1 随机欠采样
rus = RandomUnderSampler(random_state=0)

X_resampled_u1, y_resampled_u1 = rus.fit_resample(X, y)
counter_resampled_u1 = Counter(y_resampled_u1)
print("随机欠采样结果:\n", counter_resampled_u1)

plt.scatter(X_resampled_u1[:, 0], X_resampled_u1[:, 1], c=y_resampled_u1)
plt.show()

2.5、随机欠采样方法的运行结果:

2.5.png
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,332评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,508评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,812评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,607评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,728评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,919评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,071评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,802评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,256评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,576评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,712评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,389评论 4 332
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,032评论 3 316
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,798评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,026评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,473评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,606评论 2 350

推荐阅读更多精彩内容