ЛММО. 9/24.3/11.7/12.Задача 4

Получите предсказанные значения целевого признака тестовой выборки и сохраните их в переменную predictions.

Выведите второе предсказанное значение из переменной predictions и соответствующее известное значение из переменной y_test, а потом сравните их. Хорошо ли сработала модель?

# необходимые импорты
import pandas as pd
from sklearn.linear_model import LinearRegression
from sklearn.model_selection import train_test_split
# создание константы RANDOM_STATE
RANDOM_STATE = 42
# считывание CSV-файла в датафрейм pandas, сохранение выборок
df = pd.read_csv('real_estate_2.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
)
# инициализация и обучение модели
model_lr  = LinearRegression()
model_lr.fit(X_train, y_train)
# предсказанные значения для X_test
predictions = model_lr.predict(X_test)
print(predictions[1], y_test.reset_index(drop=True)[1])