Python处理样本不均衡问题

文章讨论了机器学习中常见的样本不均衡问题,及其对模型性能的影响。提出了数据重采样策略,包括上采样和下采样,以及SMOTE算法来生成新样本。此外,还介绍了PyTorch的ImbalancedDatasetSampler和权重法来调整不同类别样本的权重。文章通过示例展示了如何使用SMOTE平衡信用卡欺诈检测数据集,并进行了模型训练。

1、背景

样本类别不均衡是机器学习实践中经常会遇到的问题,很多时候我们只有少量的正样本和非常多的负样本,当然这种情况在多分类问题中也会经常遇到。

这种现象会对机器学习模型的性能带来很大影响。因为负样本很多,负样本在总的损失函数中的占比很大,使得模型会专注与学习负样本相关的信息来减小损失,对于正样本的关注度不足,导致最终模型对正样本以及与正样本相似的负样本的分类效果较差。

2、数据重采样

为了解决上述数据不均衡的问题,一个被广泛采用的技术叫做数据重采样。它包含从多数类别的样本中删除个别样本(下采样)以及往少数类别的样本中添加更多样本(过采样)。

(1)上采样

通过采样对样本进行扩充,简单的重复样本不是和好方法,最好使用数据增强的方式生成相对自然又有新信息的样本。

缺点: 实现过采样一种简单的方式是重复复制数量较少的类别的样本,这有可能会导致模型过拟合。

(2)下采样

通过随机采样、分层采样、加权采样等方式对样本进行筛选,保留比较有代表性的样本,去掉大量重复的相似的样本。

缺点: 实现下采样最简单的方式是删除数量较多的某类样本,这可能会导致训练过程信息丢失。

3、权重法

样本数量多类别,样本权重小一些;样本数量少的类别,样本权重大一些。

4、不均衡数据集采样器:PyTorch采样器ImbalancedDatasetSampler

特点:

  • 能够从不平衡的数据集采样后重新平衡类间分布
  • 能够自动估计采样时的权值
  • 避免创建新的平衡数据集
  • 当它与数据增强技术一起使用时,可以减轻过拟合

5、SMOTE方法

SMOTE算法的基本思想就是对少数类别样本进行分析和模拟,并将人工模拟的新样本添加到数据集中,进而使原始数据中的类别不再严重失衡。是基于随机过采样算法的一种改进方案。

该算法的模拟过程采用了KNN技术,模拟生成新样本的步骤如下:

  • 采样最邻近算法,计算出每个少数类样本的K个近邻;
  • 从K个近邻中随机挑选N个样本进行随机线性插值;
  • 构造新的少数类样本;
  • 将新样本与原数据合成,产生新的训练集;
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
from sklearn.preprocessing import StandardScaler
from sklearn.model_selection import train_test_split
from imblearn.over_sampling import SMOTE
from sklearn.ensemble import RandomForestClassifier
from sklearn.metrics import confusion_matrix
from sklearn.model_selection import train_test_split

# 读取数据集并处理
data = pd.read_csv("E:/file/creditcard.csv")
data<
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

rubyw

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值