Label encoding

Nu ska vi arbeta med kodning av kategoriska variabler. Du kommer återigen att använda ett delurval från Kaggle-tävlingen House Prices.

Målet är att koda de kategoriska särdragen "RoofStyle" och "CentralAir" med label encoding. DataFrames train och test finns redan tillgängliga i din miljö.

Den här övningen är en del av kursen

Vinna en Kaggle-tävling i Python

Visa kurs

Övningsinstruktioner

Konkatenera DataFrames train och test till en enda DataFrame houses.
Skapa ett LabelEncoder-objekt utan argument och tilldela det till le.
Skapa nya label-kodade särdrag för "RoofStyle" och "CentralAir" med samma le-objekt.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

# Concatenate train and test together
houses = ____.____([train, test])

# Label encoder
from sklearn.preprocessing import LabelEncoder
le = ____()

# Create new features
houses['RoofStyle_enc'] = le.fit_transform(houses[____])
houses['CentralAir_enc'] = ____.____(____[____])

# Look at new features
print(houses[['RoofStyle', 'RoofStyle_enc', 'CentralAir', 'CentralAir_enc']].head())

Redigera och kör kod