Skip to content

Latest commit

 

History

History

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 

README.md

Прогнозирование количества заказов такси

Для корректного отображения графиков, пожалуйста, перейдите по ссылке на nbviewer: Jupyter notebook on nbviewer

Задача

Требуется построить модель машинного обучения для прогнозирования количества заказов такси на следующий час на основе исторических данных.

Значение метрики RMSE на тестовой выборке должно быть <= 48.

Использовались следующие модели:

  • Градиентный бустинг (CatBoost),
  • Линейная регрессия

Был проведен ресемплинг данных по 1 часу, были изучены тренды и сезонность. Добавлены новые фичи: календарные признаки, отстающие значения и скользящее среднее.

Оптимальные параметры были найдены с помощью GridSearchCV

Данные

Дата-время и целевой признак - количество заказов.

Используемые библиотеки

statsmodels.tsa.seasonal
catboost
pandas
sklearn
numpy
matplotlib.pyplot