Ты тренер покемонов и хочешь предсказать общую силу покемона (
Total) на основе его характеристик!
Датасет содержит информацию о 721 покемоне с характеристиками:
-
HP — очки здоровья
-
Attack — сила атаки
-
Defense — защита
-
Sp. Atk — специальная атака
-
Sp. Def — специальная защита
-
Speed — скорость
-
Total — общая сила (сумма всех характеристик) — это наша цель!
-
Type 1, Type 2 — типы покемона (Fire, Water, Grass и т.д.)
-
Generation — поколение (1-6)
-
Legendary — легендарный ли покемон (True/False)
Выполни следующее
- Загрузи и подготовь данные
- Раздели данные на обучающую и тестовую выборки:
Параметры разделения:
- разметр тестовой выборки равен 20% (~144 покемона)
random_state=42
- Обучи модель
- Сделай предсказания на тестовой выборке
- Вычисли метрики качества (R2, MAE, RMSE)
- Выведи на экран значение каждой метрики (с точностью до 2-х знаков после запятой): в первой строке выведи значение R2 , во второй - MAE, в третьей - RMSE.
Для тестирования своей программы можно скачать датасет по этой
ссылке