vgrabovets
4/24/2017 - 2:06 PM

xgb_feature_importance

xgb_feature_importance

def xgb_feature_importance(model, feature_list):
    scores = model.get_fscore()
    sum_scores = sum(scores.values())
    fnames = model.feature_names
    result = {feature_list[i]: round(scores.get(fnames[i], 0) / sum_scores, 3)
              for i in range(len(feature_list))}
    return sorted(result.items(), key=lambda x: x[1], reverse=True)