Reintroducerea pădurii aleatoare

Machine Learning cu caret în R

Zach Mayer

Data Scientist at DataRobot and co-author of caret

Recapitulare: pădurea aleatoare

  • Mai lentă de antrenat decât glmnet
  • Mai puțin interpretabilă
  • Adesea (dar nu întotdeauna) mai precisă decât glmnet
  • Mai ușor de ajustat
  • Necesită preprocesare redusă
  • Captează efecte de prag și interacțiuni între variabile
Machine Learning cu caret în R

Pădure aleatoare pe datele churn

set.seed(42)
churnTrain$churn <- factor(
  churnTrain$churn, levels = c("no", "yes")
)

model_rf <- train(
  churn ~ ., 
  churnTrain,
  metric = "ROC",
  method = "ranger", 
  trControl = myControl
)
Machine Learning cu caret în R

Pădure aleatoare pe datele churn

plot(model_rf)

pasted-image-1261.png

Machine Learning cu caret în R

Să exersăm!

Machine Learning cu caret în R

Preparing Video For Download...