Дерево сюжета xgboost выдает ошибку в R - PullRequest
0 голосов
/ 09 марта 2019

Я запустил модель 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 xgb.plot.tree code error

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...