[問題] xgboost模型可視化

作者: interwise613 (BG蔚)   2020-07-16 09:43:52
想把訓練好的xgb模型畫出樹來看結構
但用xgb.to_graphviz畫出來的樹看起來醜醜的
有辦法把訓練好的模型用sklearn.tree裡的export_graphviz畫圖嗎?主要是想讓每個節
點預測值大小用不同顏色呈現,看起來較一目了然
作者: Starcraft2 (來自星海的你)   2020-07-16 13:47:00
xgb.to_graphviz會得到graphviz.Source() object感覺就可以用pydotplus的標準api去處理?喔等等好像懂你遇到的問題了 我試試看先

Links booklink

Contact Us: admin [ a t ] ucptt.com