[問題] lightgbm 的 plot_tree()
作業系統:(ex:mac,win10,win8,win7,linux,etc...)
win10
問題類別:(ex：ML,SVM,RL,DL,RNN,CNN,NLP,BD,Vis,etc...)
ML
使用工具:(ex:python,Java,C++,tensorflow,keras,MATLAB,etc...)
python
問題內容:
各位大大前輩大家好,
小弟用 python 的 lightgbm trian 完 model 後,
使用 plot_tree 畫決策樹,
大致如下:
gbm = lgb.train(params,
lgb_train,
num_boost_round=100,
valid_sets=[lgb_train, lgb_test],
evals_result=evals_result,
verbose_eval=10)
lgb.plot_tree(gbm, tree_index=3, figsize=(20, 8), show_info=['split_gain',
'internal_value', 'internal_count', 'internal_weight', 'leaf_count',
'leaf_weight', 'data_percentage'])
plt.show()
小弟的問題是
1. 請問畫出來 tree 中的 leaf, 左邊還是右邊才是 target = 1
2. 圖中的 internel_value 是什麼意思?
3. 原本以為 leaf 為左邊是 target = 1,
servey 後, 感覺 leaf_value 是未經 sigmoid 的預測值.
但我畫出來的 tree 左邊的 leaf 的 leaf_value 有負,
所以經 sigmoid 後為小於 0.5, 應該為 target = 0.
希望有經驗的大大們給予一些解惑,
乾蝦~
--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 1.163.208.251 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/DataScience/M.1587477611.A.864.html
→
05/12 02:18,
4年前
, 1F
05/12 02:18, 1F
→
05/12 02:18,
4年前
, 2F
05/12 02:18, 2F
DataScience 近期熱門文章
PTT數位生活區 即時熱門文章
