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值。