Я запустил модель xgboost и пытаюсь построить деревья, но формула дает мне ошибку.
Модель данных обучения X-gboost:
tuneGridXGB <- expand.grid(
nrounds=c(150),
max_depth = c(10,15,25,35),
eta = 0.05,
gamma = c(0.1, 1),
colsample_bytree = c(0.5,0.75),
subsample = c(0.50, 0.75),
min_child_weight = c(2,5))
cv_xgboost <- caret::train(as.factor(status_group)~.,
data=data.matrix(training_data),
method = 'xgbTree',
metric = 'Accuracy',
trControl = train_control,
tuneGrid = tuneGridXGB)
cv_xgboost$results
Я использую эту формулу для построения дерева:
xgb.plot.tree(feature_names = cv_xgboost$finalModel$coefnames,
model = cv_xgboost$finalModel,
trees = 1,
render = FALSE,
show_node_id = FALSE)
и это дает мне следующую ошибку:
Ошибка кода xgb.plot.tree