![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
gbdt
aone322
这个作者很懒,什么都没留下…
展开
-
lightgbm画决策树方法
lightgbm画树的两种方法,画树前需要安装graphviz画图工具包pip install graphviz安装后运行画图代码报错,找不到grahviz的dot什么的,因为graphviz不仅要在pyhton环境中安装,需要同时在系统安装graphviz:Linux下安装方法为:$sudo yum install graphviz一、使用plot_treefig = plt.figure(figsize=(200, 200))ax = fig.subplots()lgb.plot_原创 2020-09-17 11:43:26 · 3764 阅读 · 0 评论 -
LGB交叉验证KFOLD
# 引入相关包import pandas as pdimport lightgbm as lgbfrom sklearn.model_selection import StratifiedKFold# 假设这里准备好了训练数据train_data,它是一个pandas的dataframe,包括特征列和score列train_label = train_data['score']# 初始化一个k-fold生成器NFOLDS = 5kfold = StratifiedKFold(n_spli原创 2020-09-15 17:45:34 · 1298 阅读 · 0 评论