Metrici de performanță pentru arbori de regresie

Machine Learning cu modele bazate pe arbori în R

Sandro Raabe

Data Scientist

Cum se măsoară performanța?

  • Probleme de clasificare: acuratețe (matrice de confuzie)
  • Probleme de regresie: „corect" este relativ, fără corectitudine binară

$\Rightarrow$ Se măsoară cât de departe sunt predicțiile față de valoarea reală

Machine Learning cu modele bazate pe arbori în R

Metrici comune pentru regresie

  • Eroarea Absolută Medie (MAE)
  • Rădăcina Erorii Pătratice Medii (RMSE)

 

Intuiția MAE:

grafic al diferențelor medii

 

 

 

 

 

MAE = lungimea medie a barelor roșii

Machine Learning cu modele bazate pe arbori în R

Formule și intuiție

 

$$MAE = \frac{1}{n} \sum_{i=1}^n\left| actual_i - predicted_i \right|$$

 

  • „Suma abaterilor absolute împărțită la numărul de predicții"

$$\quad MSE = \quad \frac{1}{n} \sum_{i=1}^n\left( actual_i - predicted_i \right)^2$$

  •                   „Eroarea pătratică medie"
Machine Learning cu modele bazate pe arbori în R

Formule și intuiție

 

$$MAE = \frac{1}{n} \sum_{i=1}^n\left| actual_i - predicted_i \right|$$

 

  • „Suma abaterilor absolute împărțită la numărul de predicții"

$$RMSE = \sqrt{\frac{1}{n} \sum_{i=1}^n\left( actual - predicted \right)^2}$$

  • „Rădăcina erorii pătratice medii"
  • Erorile mari primesc o pondere mai mare
Machine Learning cu modele bazate pe arbori în R

Cod: predicții

# parsnip and yardstick are included in tidymodels
library(tidymodels)

# Make predictions and add to test data
predictions <- predict(model, new_data = chocolate_test) %>%

  bind_cols(chocolate_test)

# A tibble: 358 x 7
   .pred final_grade review_date cocoa_percent company_location
   <dbl>       <dbl>       <int>         <dbl> <fct>           
 1  2.5         2.75        2013          0.7  France          
 2  3.64        3.25        2014          0.8  France          
 3  3.3         3.5         2012          0.7  France          
 4  3.25        3.5         2011          0.72 Fiji            
# ... with 354 more rows, and 2 more variables: bean_type <fct>, broad_bean_origin <fct>
Machine Learning cu modele bazate pe arbori în R

Cod: mae() și rmse()

# Evaluate using mae()
mae(predictions,

    estimate = .pred,

    truth = final_grade)

# A tibble: 1 x 2
  .metric   .estimate
  <chr>         <dbl>
1 mae           0.363

# Evaluate using rmse()
rmse(predictions,
     estimate = .pred,
     truth = final_grade)

# A tibble: 1 x 2
  .metric   .estimate
  <chr>         <dbl>
1 rmse          0.457
Machine Learning cu modele bazate pe arbori în R

Hai să evaluăm!

Machine Learning cu modele bazate pe arbori în R

Preparing Video For Download...