SKlearn GBM功能重要性或“预测”方法的贡献者

时间:2019-01-21 12:08:22

标签: python machine-learning scikit-learn

我正在使用SKlearn GBM预测进行一次练习,并且为了在拟合训练数据后理解特征重要性,我可以在python中轻松地做到这一点,因为'fit'方法具有这些功能

但是我也想知道功能对测试数据集的重要性,但是'predict'方法没有这样的东西

from sklearn import ensemble
gbm = ensemble.GradientBoostingRegressor(**params)## 
gbm.fit(X_train, y_train))
# feature importance 
feat_imp = pd.DataFrame(gbm.feature_importances_)

有没有解决方案,可以帮助我了解测试的重要功能或使用sklearn gbm或其他方式预测数据集

感谢所有帮助!

0 个答案:

没有答案