python - 从 imblearn 导入 RUSBoostClassifier 的问题

Question

我一直坚持按照这个例子导入 RUSBoostClassifier

from imblearn.ensemble import RUSBoostClassifier

我收到以下错误：

ImportError                               Traceback (most recent call last)
<ipython-input-12-d0ff5157a81a> in <module>
----> 1 from imblearn.ensemble import RUSBoostClassifier
  2 from sklearn.datasets import make_classification
  3 
  4 X, y = make_classification(n_samples=1000, n_classes=3,
  5                            n_informative=4, weights=[0.2, 0.3, 0.5],

ImportError: cannot import name 'RUSBoostClassifier'

我不明白为什么！我可以轻松使用 imblearn 中的其他模块，例如

from imblearn.under_sampling import RandomUnderSampler

或者

from imblearn.over_sampling import SMOTE 

我用的是 Jupyter notebook，Python 版本是 3.6.6。我已经更新了 sklearn 包并按照此链接重新安装了 imblearn 包。

有没有人有任何想法来解决这个问题？或者有想法在代码中直接使用这个类（源代码）？

Answer 1

问题是imblearn包的版本。0.4.3 版支持imblearn 的新模块，例如RUSBoostClassifieror BalancedRandomForestClassifier。imblearn可以检查包的版本!pip list（感谢@desertnaut）。如果您使用的是旧版本，请将其完全删除（重新安装 imblearn 实际上对我没有帮助）。之后按照此说明安装它。在这种情况下，最新版本的 imblearn 将添加到您的 anaconda 环境中。