MAE на train и test: сопоставьте прогнозы с правильными ответами

MAE на обучающих и отложенных строках

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

Что уже дано

В редакторе — цельный скрипт: загрузка diabetes, совместное разбиение с test_size=0.2, random_state=0, обучение одной LinearRegression на train и два вычисления MAE. Скрипт выполняется, но тестовая ошибка сравнивает прогнозы не с ответами соответствующих строк. Скрытая подготовка не создаёт данные.

Что нужно сделать

Исправьте сопоставление прогнозов и истинных ответов, сохранив обучение только на train. Итогом должны быть mae_train и mae_test для одной модели; не округляйте их. Другие части датасета и выборки не меняйте.

Ввод и вывод

stdin не используется. Печатать через print в stdout ничего не нужно. Проверяются верхнеуровневые X_train, X_test, y_train, y_test, model, pred_train, pred_test, mae_train, mae_test.