Случайный лесной пакет в R

Question

Я использую случайный лесной пакет в R для регрессии, он дает мне два вида информации: Mean of squared residuals и% Var объяснил. Но я хочу вычислить RMSE а также R^2 из обучающих и тестовых наборов, кто-нибудь может мне помочь, как я могу найти такую ​​информацию?

Answer 1

Извините, это не конкретный ответ, но у меня недостаточно средств, чтобы оставить комментарий.

Трудно сказать, как вы можете получить то, что хотите, без воспроизводимого примера. Однако, если вы использовали xtest= а также ytest= аргументы в призыве к randomForest (при условии, что вы используете пакет randomForest), то то, что вы ищете, должно быть частью результирующего объекта randomForest. То, что вы хотите посмотреть, это test часть результирующего списка случайных лесов.

Попытка примера:

rf.results <- randomForest( whatever arguments )
rf.results$test$mse  # mse (maybe you can take the square root to get rmse)
rf.results$test$rsq  # pseudo-R2 for random forest

Если у вас загружен случайный пакет леса, вы можете проверить эту информацию, а также провести некоторые исследования самостоятельно с ?randomForest, Раздел "Значение" документации детализирует объект, который является результатом вызова randomForest и где вы можете найти различные показатели производительности.