首页 > 解决方案 > 使用 smote 时出现此错误

问题描述

在此处输入图像描述

# apply SMOTE to tackle class imbalance
from imblearn.over_sampling import SMOTE
sm = SMOTE(kind = "regular", k=1)
X_tr,y_tr = sm.fit_sample(X_train,y_train)
print(X_tr.shape)
print(y_tr.shape)

请帮我解决这个问题

标签: pythonimblearnsmote

解决方案


SMOTE就文档而言,过采样算法没有kind您建议的参数。

类 imblearn.over_sampling.SMOTE(*, sampling_strategy='auto', random_state=None, k_neighbors=5, n_jobs=None)

如果您的意思是对、do、等的过采样,则minoritynot minority参数是 thesampling_strategy并且默认为auto

sm = SMOTE(sampling_strategy = "minority")

推荐阅读