Feature Importance with XGBClassifier

后端 未结 9 1166
隐瞒了意图╮
隐瞒了意图╮ 2020-12-14 10:03

Hopefully I\'m reading this wrong but in the XGBoost library documentation, there is note of extracting the feature importance attributes using feature_importances_

9条回答
  •  暖寄归人
    2020-12-14 10:58

    Get Feature Importance as a sorted data frame

    import pandas as pd
    import numpy as np
    def get_xgb_imp(xgb, feat_names):
        imp_vals = xgb.booster().get_fscore()
        feats_imp = pd.DataFrame(imp_vals,index=np.arange(2)).T
        feats_imp.iloc[:,0]= feats_imp.index    
        feats_imp.columns=['feature','importance']
        feats_imp.sort_values('importance',inplace=True,ascending=False)
        feats_imp.reset_index(drop=True,inplace=True)
        return feats_imp
    
    feature_importance_df = get_xgb_imp(xgb, feat_names)
    

提交回复
热议问题