Tillämpa vtreat på cykeluthyrningsdata
I den här övningen skapar du one-hot-kodade dataramar av cykeldata för juli och augusti, som sedan används med xgboost.
Dataramarna bikesJuly och bikesAugust har laddats in i förväg.
För enkelhets skull har variabeln vars definierats med listan över variabelkolumner för modellen.
Den här övningen är en del av kursen
Övervakad inlärning i R: Regression
Övningsinstruktioner
- Ladda paketet
vtreat. - Använd
designTreatmentsZ()för att skapa en behandlingsplantreatplanför variablerna ivarsfrånbikesJuly(träningsdata).- Sätt flaggan
verbose=FALSEför att förhindra att funktionen skriver ut för många meddelanden.
- Sätt flaggan
- Fyll i luckorna för att skapa en vektor
newvarssom bara innehåller namnen på de transformerade variablerna av typencleanochlev. Skriv ut den. - Använd
prepare()för att skapa en one-hot-kodad träningsdatarambikesJuly.treat.- Använd argumentet
varRestrictionsför att begränsa variablerna tillnewvars.
- Använd argumentet
- Använd
prepare()för att skapa en one-hot-kodad testdatarambikesAugust.treatfrånbikesAugustpå samma sätt. - Anropa
str()på båda de förberedda testramarna för att se deras struktur.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# The outcome column
(outcome <- "cnt")
# The input columns
(vars <- c("hr", "holiday", "workingday", "weathersit", "temp", "atemp", "hum", "windspeed"))
# Load the package vtreat
___
# Create the treatment plan from bikesJuly (the training data)
treatplan <- ___(___, ___, verbose = FALSE)
# Get the "clean" and "lev" variables from the scoreFrame
(newvars <- treatplan %>%
use_series(scoreFrame) %>%
filter(code %in% ___) %>% # get the rows you care about
use_series(___)) # get the varName column
# Prepare the training data
bikesJuly.treat <- ___(___, ___, varRestriction = ___)
# Prepare the test data
bikesAugust.treat <- ___(___, ___, varRestriction = ___)
# Call str() on the treated data
___
___