1

構築されたツリーで実際に使用されている変数を知るにはどうすればよいですか?

model = tree(status~., set.train)

次のように書くと、変数を見ることができます。

summary(model)

tree(formula = status ~ ., data = set.train)
Variables actually used in tree construction:
[1] "spread1"      "MDVP.Fhi.Hz." "DFA"          "D2"           "RPDE"                "MDVP.Shimmer" "Shimmer.APQ5"
Number of terminal nodes:  8 
Residual mean deviance:  0.04225 = 5.831 / 138 
Distribution of residuals:
   Min. 1st Qu.  Median    Mean 3rd Qu.    Max. 
-0.9167  0.0000  0.0000  0.0000  0.0000  0.6667

しかし、実際に使用されている変数のインデックスであるベクトルを取得するにはどうすればよいですか?

4

4 に答える 4

1

これがあなたが探しているものだと思います

fit <- rpart(Species ~ Sepal.Length + Sepal.Width + Petal.Length + Petal.Width, data=iris) used.var <- setdiff(levels(fit$frame$var), "<leaf>")

于 2019-10-19T23:52:34.203 に答える
-2

同様のパッケージに切り替えたい場合rpartは、使用されている変数を重要度順に並べ替えてから直接取得できます。fit

fit <- rpart(Species ~., data=iris)
fit$variable.importance
于 2016-03-31T08:26:56.570 に答える