浅墨゛ღ
2022-06-30 16:38
浏览 13

深度学习数据处理样本均衡问题

深度学习随机下采样中的X为高维数据,不能直接用randomundersampler.fit_resample(X,Y)
代码

from imblearn.under_sampling import RandomUnderSampler # 样本均衡 欠采样
rus = RandomUnderSampler(random_state=0)
temp1, temp2 = rus.fit_resample(Y, Y)
index_rus = rus.sample_indices_#获取均衡后的索引
train_data = X[index_rus]#根据索引取数据
train_label = Y[index_rus]#根据索引取标签

问题

rus.fit_resample(Y, Y),这里传入的是两个二维的数据标签Y,可以这样用吗?

相关推荐 更多相似问题