2016-03-12 5 views
7

xgboostが私の木をプロットしていないのはなぜですか?次のように私は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スクリプトはありません。私は重要度マトリックスをプロットするようにすべきだと考えていました。

私には何が欠けていますか?

+0

私と同じ問題です。 – joscani

+0

あなたの書式とタイプミスを修正しなければなりませんでした。コードではブロック書式ではなくコード書式を使用してください。また、これはRStudioの問題ではなく、Rで再実行することでテストできました。 – smci

+0

また、Rとxgboostのバージョン:4.4にいくつかの既知の問題があった場合にも役立ちます。 – smci

答えて

0

ツリーにノードが1つしかない場合はプロットされません。これは最初の2つのツリーの場合です。最初にxgb.dumpでツリーをダンプし、複数のノードを持つツリーを確認し、それに応じてn_first_treeの値を増やすことができます。

関連する問題