1. Learn
    2. /
  2. Courses
    3. /
  3. Machine Learning з PySpark
Connected

Exercise

Крос-валідація простого моделя тривалості перельоту

Ви вже побудували кілька моделей для прогнозування тривалості перельоту та оцінили їх за допомогою простого поділу на тренувальну й тестову вибірки. Однак крос-валідація дає значно кращий спосіб оцінити якість моделі.

У цій вправі ви натренуєте просту модель тривалості перельоту з використанням крос-валідації. Час у дорозі зазвичай сильно корелює з відстанню, тож одна лише колонка km має дати непогану модель.

Дані випадково поділено на flights_train і flights_test.

Наступні класи вже імпортовано: LinearRegression, RegressionEvaluator, ParamGridBuilder і CrossValidator.

Instructions

100 XP
  • Створіть порожню ґратку параметрів.
  • Створіть об'єкти для побудови й оцінювання моделі лінійної регресії. Модель має передбачати поле "duration".
  • Створіть об'єкт крос-валідації. Задайте значення аргументів estimator, estimatorParamMaps та evaluator. Оберіть 5-кратну крос-валідацію.
  • Натренуйте й протестуйте модель на кількох фолдах тренувальних даних.