Kodowanie etykietami

Czas popracować z kodowaniem zmiennych kategorycznych. Znowu skorzystasz z podzbioru danych z konkursu Kaggle House Prices.

Twoim zadaniem jest zakodowanie cech kategorycznych "RoofStyle" i "CentralAir" za pomocą kodowania etykietami. DataFramy train i test są już dostępne w twoim środowisku pracy.

To ćwiczenie jest częścią kursu

Zwycięstwo w konkursie Kaggle w Pythonie

Zobacz kurs

Instrukcje do ćwiczenia

Połącz DataFramy train i test w jeden DataFrame houses.
Utwórz obiekt LabelEncoder bez argumentów i przypisz go do zmiennej le.
Stwórz nowe cechy zakodowane etykietami dla kolumn "RoofStyle" i "CentralAir", używając tego samego obiektu le.

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

# 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())

Edytuj i uruchom kod