从imblearn导入RUSBoostClassifier的问题

时间:2018-11-13 13:08:03

标签: python machine-learning anaconda python-import imblearn

this example之后,我一直无法导入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笔记本,Python版本是3.6.6。我已根据此link更新了sklearn软件包并重新安装了imblearn软件包。

有人有解决这个问题的想法吗?还是有想法在代码中直接使用此类(源代码)?

1 个答案:

答案 0 :(得分:0)

问题是imblearn软件包的版本。版本0.4.3支持新的imblearn模块,例如RUSBoostClassifierBalancedRandomForestClassifierimblearn软件包的版本可以通过!pip list进行检查(感谢@desertnaut)。如果您使用的是旧版本,请将其完全删除(重新安装imblearn实际上对我没有帮助)。之后,按照this instruction进行安装。在这种情况下,最新版本的imblearn将添加到您的anaconda环境中。