Кероване навчання в R: регресія
Nina Zumel and John Mount
Win-Vector, LCC
Кілька різних дерев рішень, усереднених разом
cnt ~ hr + holiday + workingday +
weathersit + temp + atemp + hum + windspeed

model <- ranger(fmla, bikesJan,
num.trees = 500,
respect.unordered.factors = "order")
formula, datanum.trees (типово 500) — використовуйте щонайменше 200mtry — кількість змінних для спроби в кожній вершиніrespect.unordered.factors — радимо встановити "order"model
Ranger result
...
OOB prediction error (MSE): 3103.623
R squared (OOB): 0.7837386
Алгоритм random forest повертає оцінки позавибіркової якості.
bikesFeb$pred <- predict(model, bikesFeb)$predictions
Вхідні дані для predict():
Прогнози доступні в елементі predictions.
Обчисліть RMSE:
bikesFeb %>%
mutate(residual = cnt - pred) %>%
summarize(rmse = sqrt(mean(residual^2)))
rmse
1 67.15169
| Модель | RMSE |
|---|---|
| Quasipoisson | 69.3 |
| Random forests | 67.15 |


Кероване навчання в R: регресія