【发布时间】:2020-04-27 19:58:47
【问题描述】:
我正在使用 SMOTE 函数对包含大约 98% 0 和 2% 1 的稀疏数据集进行过采样。我使用了以下代码
from imblearn.over_sampling import SMOTE
import os
import pandas as pd
df_input= pd.read_csv('input_tr.csv',index_col=0)
train_X=df_input.ix[:, df_input.columns != 'row_num']
df_output=pd.read_csv("output_tr.csv",index_col=0)
train_y=df_output
sm = SMOTE(random_state=12, ratio = 1.0)
train_X_sm,train_y_sm=sm.fit_sample(train_X,train_y)
我收到以下错误
line 347, in kneighbors
(train_size, n_neighbors)
ValueError: Expected n_neighbors <= n_samples, but n_samples = 4, n_neighbors = 6
你能帮我解决这个错误吗?
【问题讨论】: