[問題] xgboost模型可視化

看板Python作者 (BG蔚)時間3年前 (2020/07/16 09:43), 編輯推噓2(2推 0噓 1→)
留言3則, 1人參與, 3年前最新討論串1/1
想把訓練好的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
推
07/16 13:47, 3年前 , 1F
xgb.to_graphviz會得到graphviz.Source() object
07/16 13:47, 1F
推
07/16 13:51, 3年前 , 2F
感覺就可以用pydotplus的標準api去處理?
07/16 13:51, 2F
→
07/16 13:52, 3年前 , 3F
喔等等好像懂你遇到的問題了 我試試看先
07/16 13:52, 3F
‣ 更多 interwise613 的文章
文章代碼(AID): #1V3x3Qiy (Python)