如何可视化BorutaPy(Python实现的Boruta算法)提取的特征结果?

时间:2019-04-25 08:14:19

标签: python-3.x scikit-learn

我正在使用Boruta从数据中提取要素。现在,我已经使用BorutaPy(rf, n_estimators='auto', verbose=3, random_state=1).fit(X_std,y)提取了11个特征,但是我不知道如何可视化结果?

我用的是:

rf = RandomForestClassifier(n_jobs=-1, class_weight='balanced', max_depth=5)
feat_selector = BorutaPy(rf, n_estimators='auto', verbose=3, random_state=1)
feat_selector.fit(X_std, y)
feat_selector.support_

我从数据中获得11个功能。但是我想用R可视化结果类似plot(Boruta.mydata)。我不知道如何使用python来实现它。

下面的图片显示了我想要的结果:

enter image description here

1 个答案:

答案 0 :(得分:0)

我最近也遇到了这个问题-Python无法绘制Boruta图。 我最终使用R来完成它,并将重要功能导出回Python