R: xgboost перекрестная проверка замораживания / завершения сеанса RStudio - PullRequest
0 голосов
/ 28 апреля 2020

Я борюсь с выполнением перекрестной проверки xgboost на относительно небольшом наборе данных, но следующая команда последовательно убивает мой сеанс RStudio:

xgb.cv(data = X_train_dmat, 
        eta = 0.1,
        max_depth = 5, 
        nround=25, 
        subsample = 0.5,
        colsample_bytree = 0.5,
        seed = 1,
        booster = 'gbtree',
        folds=foldsCV,
        objective = 'reg:squarederror',
        nthread = 3)

Обратите внимание, что следующее работает для того же набора данных:

xgb <- xgboost(data = X_train_dmat, 
               eta = 0.1,
               max_depth = 5, 
               nround=25, 
               subsample = 0.5,
               colsample_bytree = 0.5,
               seed = 1,
               booster = 'gbtree',
               objective = 'reg:squarederror',
               nthread = 3
)

Я пытаюсь регрессировать как с числовыми, так и с категориальными переменными. Какие шаги я могу go устранить неисправности? Существуют ли очевидные настройки параметров, которые могут вызвать проблемы?

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