Měření výkonu křížové validace

Machine Learning in the Tidyverse

Dmitriy (Dima) Gorenshteyn

Lead Data Scientist, Memorial Sloan Kettering Cancer Center

Měření výkonu

Měření výkonu – skutečnost

Měření výkonu – predikce

Měření výkonu

Střední absolutní chyba

Složky měření výkonu

1) Skutečné hodnoty life_expectancy
2) Predikované hodnoty life_expectancy
3) Metrika pro porovnání 1) a 2)

1) Extrakce skutečných hodnot

cv_prep_lm <- cv_models_lm %>% 
  mutate(validate_actual = map(validate, ~.x$life_expectancy))

Funkce predict() a map2()

predict(model, data)

map2(.x = model, .y = data, .f = ~predict(.x, .y))

2) Příprava predikovaných hodnot

cv_prep_lm <- cv_eval_lm %>% 
  mutate(validate_actual = map(validate, ~.x$life_expectancy),
         validate_predicted = map2(model, validate, ~predict(.x, .y)))

3) Výpočet MAE

library(Metrics)
cv_eval_lm <- cv_prep_lm %>% 
  mutate(validate_mae = map2_dbl(validate_actual, validate_predicted, 
                                ~mae(actual = .x, predicted = .y)))

cv_eval_lm

#  5-fold cross-validation 
# A tibble: 5 x 8
splits       id    train validate model validate_a. validate_p validate_mae
<S3: rsplit> Fold1 <tib. <tib.   <S3.   <dbl.       <dbl.       1.47
<S3: rsplit> Fold2 <tib. <tib.   <S3.   <dbl.       <dbl.       1.51
<S3: rsplit> Fold3 <tib. <tib.   <S3.   <dbl.       <dbl.       1.44
<S3: rsplit> Fold4 <tib. <tib.   <S3.   <dbl.       <dbl.       1.48
<S3: rsplit> Fold5 <tib. <tib.   <S3.   <dbl.       <dbl.       1.68

Pojďme si procvičit!

Machine Learning in the Tidyverse