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