4

我正在使用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
4

1 回答 1

8

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

于 2019-08-13T11:17:08.970 回答