您的位置:首页 > 其它

xgboost 保存模型和特征重要度

2017-10-16 21:44 1731 查看
1. 保存模型:

(1)保存:

bst.save_model('0001.model')
(2)加载模型:

bst_new = xgb.Booster({'nthread':4}) #init model
bst_new.load_model("0001.model") # load data


source from:
http://blog.csdn.net/zc02051126/article/details/46771793
2.xgb 打印特征重要度:

使用f1 score来打分

import pandas as pd
import matplotlib.pylab as plt
feat_imp = pd.Series(clf.booster().get_fscore()).sort_values(ascending=False)
feat_imp.plot(kind='bar', title='Feature Importances')
plt.ylabel('Feature Importance Score')
plt.show()来自,余音大神的blog: http://blog.csdn.net/q383700092/article/details/53763328
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐