У цій вправі ви скористаєтеся vtreat, щоб виконати one-hot-encoding категоріальної змінної на невеликому прикладі. vtreat створює план обробки для перетворення категоріальних змінних на індикаторні змінні (позначені кодом "lev"), а також для очищення хибних значень у числових змінних (позначені кодом "clean").

Щоб спроєктувати план обробки, використайте функцію designTreatmentsZ() (docs)

treatplan <- designTreatmentsZ(data, varlist)

data: початковий тренувальний датафрейм
varlist: вектор вхідних змінних, які потрібно обробити (рядки).

designTreatmentsZ() повертає список з елементом scoreFrame: це датафрейм, що містить назви та типи нових змінних:

scoreFrame <- treatplan %>% 
            magrittr::use_series(scoreFrame) %>% 
            select(varName, origName, code)

varName: назва нової обробленої змінної
origName: назва початкової змінної, з якої походить оброблена змінна
code: тип нової змінної.
- "clean": числова змінна без NA або NaN
- "lev": індикаторна змінна для певного рівня початкової категоріальної змінної.

(magrittr::use_series() (docs) — це синонім до $, який можна використовувати в конвеєрах.)

Для цієї вправи нам потрібні varName, де code — "clean" або "lev":

newvarlist <- scoreFrame %>% 
             filter(code %in% c("clean", "lev") %>%
             magrittr::use_series(varName)

Щоб перетворити набір даних на повністю числові змінні з one-hot-encoding, скористайтеся prepare() (docs):

data.treat <- prepare(treatplan, data, varRestrictions = newvarlist)

treatplan: план обробки
data: датафрейм для обробки
varRestrictions: змінні, які мають бути в оброблених даних

Датасет dframe і пакет magrittr вже завантажені.

Gradient Boosting-машини

Create Your Free Account