R Caret Рассчитать RMSE для комплекта

Question

Мне интересно, как я могу рассчитать RMSE для набора тестирования.

Я использовал приведенный ниже код для обучения модели:

model_gbm_important<-train(trainSetSmall[,predictors_gbm],trainSetSmall[,outcomeName],method='gbm', trControl=fitControl)

Я могу получить производительность модели с помощью

print(model_gbm_important)

Однако эта производительность основанана перекрестной проверке. Если у меня есть другой набор тестирования, как я могу использовать набор тестирования для оценки модели, отметьте значение RMSE?

Спасибо!

Answer 1

Для расчета СКО прогнозирования данных испытаний:

predictions <- predict(model, test_data)
RMSE(predictions, test_data$outcomeName)