作者interwise613 (BG蔚)
看板Python
标题[问题] xgboost模型可视化
时间Thu Jul 16 09:43:52 2020
想把训练好的xgb模型画出树来看结构
但用xgb.to_graphviz画出来的树看起来丑丑的
有办法把训练好的模型用sklearn.tree里的export_graphviz画图吗？主要是想让每个节
点预测值大小用不同颜色呈现，看起来较一目了然
--
※ 发信站: 批踢踢实业坊(ptt.cc), 来自: 39.11.0.201 (台湾)
※ 文章网址: https://webptt.com/cn.aspx?n=bbs/Python/M.1594863834.A.B3C.html
1F：推 Starcraft2: xgb.to_graphviz会得到graphviz.Source() object 07/16 13:47
2F：推 Starcraft2: 感觉就可以用pydotplus的标准api去处理? 07/16 13:51
3F：→ Starcraft2: 喔等等好像懂你遇到的问题了 我试试看先 07/16 13:52