为什么xgboost没有绘制我的树？

Question

我正在运行xgboost模型如下:

bst <- xgb.train(data=dtrain, booster="gbtree", objective="reg:linear",
    max.depth=5, nround=20, watchlist=watchlist,min_child_weight=10)
importance_matrix <- xgb.importance(names, model = bst)
xgb.plot.importance(importance_matrix[1:10,])

可变重要性矩阵很好地绘制,但是当我做以下时

xgb.plot.tree(feature_names = names, model = bst, n_first_tree = 2)

RStudio打开一个新的浏览器窗口并显示大量HTML,但没有图像.HTML具有创建图形等所需的所有细节,但我没有这些java脚本,我认为它应该像绘制重要性矩阵一样工作.

我错过了什么？

Answer 1

如果一棵树只有一个节点，那么它将不会被绘制，前两棵树就是这种情况。您可以首先通过转储树xgb.dump并查看哪些树具有多个节点，并n_first_tree相应地增加该值。