這是一棵用R右下角輸出時會出現的樹
非常地不清楚,無法閱讀,而且決策樹的xgb.plot.tree()有別於一般的作圖,無法開啟windows()並畫在上面(順帶一提關閉是dev.off()我找了一陣子)
所以就要用
export_graph(gr, 'tree.pdf', width=1500, height=1900)
直接打上副檔名就可以了,變成PDF檔就可以用google瀏覽器閱讀了
有一個很可能會遇上的bug我找了一段時間才知道為什麼畫不出來.
會出現
Error: export_graph() REASON:
這時候就要加上render=FALSE就可以囉(一定要加,要不然跑不出來)
檔案會在預設的位置,通常是在本機->文件底下
xgb.plot.tree(feature_names=names(xdata_train__),render=FALSE,model=result, trees=2)
iThome鐵人賽
看影片追技術