python - RandomUnderSampler' 对象没有属性 'fit_resample'

Question

我正在使用RandomUnderSamplerfrom imblearn，但出现以下错误。有任何想法吗？谢谢

from imblearn.under_sampling import RandomUnderSampler
print('Initial dataset shape %s' % Counter(y.values.squeeze()))
rus = RandomUnderSampler(random_state=42)
X_undersampled, y_undersampled = rus.fit_resample(X, y)
y_undersampled = y_undersampled.squeeze()

输出：

Initial dataset shape Counter({0: 2499739, 1: 1558})

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-18-4fe9dcfbd68e> in <module>
      1 print('Initial dataset shape %s' % Counter(y.values.squeeze()))
      2 rus = RandomUnderSampler(random_state=42)
----> 3 X_undersampled, y_undersampled = rus.fit_resample(X, y)
      4 y_undersampled = y_undersampled.squeeze()
      5 

AttributeError: 'RandomUnderSampler' object has no attribute 'fit_resample'

我正在使用的主要库：

imbalanced-learn==0.3.3
pandas==0.24.2
numpy==1.15.4
scikit-learn==0.19.2

标签： pythonjupyter-notebookjupyterimblearn

Answer 1

该方法fit_resample最近被引入imbalanced-learnAPI。要么更新imbalanced-learn，要么fit_sample改用。