推 Starcraft2: xgb.to_graphviz會得到graphviz.Source() object 07/16 13:47
推 Starcraft2: 感覺就可以用pydotplus的標準api去處理? 07/16 13:51
→ Starcraft2: 喔等等好像懂你遇到的問題了 我試試看先 07/16 13:52
想把訓練好的xgb模型畫出樹來看結構
但用xgb.to_graphviz畫出來的樹看起來醜醜的
有辦法把訓練好的模型用sklearn.tree裡的export_graphviz畫圖嗎？主要是想讓每個節
點預測值大小用不同顏色呈現，看起來較一目了然
--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 39.11.0.201 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/Python/M.1594863834.A.B3C.html