为什么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值。