ЛММО. 9/24.6/11.4/7.Задача 2

Кодируйте категориальные признаки в тренировочных данных с помощью get_dummies(). Дамми-ловушку устранять не нужно. Выведите на экран подготовленные данные.

import pandas as pd
from sklearn.model_selection import train_test_split
# создание константы RANDOM_STATE
RANDOM_STATE = 42
# добавляем данные из CSV-файла в датафрейм pandas, делим датасет на выборки
df = pd.read_csv('real_estate_clean_cat.csv')
X = df.drop('price', axis=1)
y = df['price']
X_train, X_test, y_train, y_test = train_test_split(
    X, 
    y, 
    random_state=RANDOM_STATE
)
# кодируйте данные методом get_dummies(). 
df_ohe = pd.get_dummies(X_train)
# выведите пять первых строк из датафрейма на экран
print(df_ohe.head(5))