[问题] 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