Как получить MSE для модели случайной лесной регрессии на тестовом наборе данных - PullRequest
0 голосов
/ 12 июля 2019

У меня есть модель случайной лесной регрессии, обученная на наборе данных обучения.Я хочу использовать его в тестовом наборе данных и получить MSE леса.Как я могу это сделать?

Моя модель такова:

fit = randomForest(out.cut$Vir_factor_freq ~ .,out.cut[,-1], importance = TRUE, ntree = 700, replace = TRUE, mtry = 7)

Где out.cut - данные тренировки.Я использовал:

pred = predict(fit, out.cut, type="response", OOB = TRUE)

Где out.cut - это тестовые данные, все хорошо, я получаю прогнозы, но хочу знать точность точно так же, как вы можете извлечь MSE из модели соответствия с помощью соответствия$ mse и изолируйте i-е значение.

...