Здравствуйте.
Прочтите (
Как устроен краткосрочный прогноз на Яндекс.Пробках), возможно появятся новые идеи для оптимизации идеи.
Для разработки прогнозных моделей при наличии временных рядов наиболее популярными являются модели авторегрессии, проинтегрированного скользящего среднего (ARIMA) (для одномерных временных рядов с трендом и без сезонных компонентов) и SARIMA (с трендовыми и / или сезонными компонентами).
Ссылки:
1.
statsmodels.tsa.statespace.sarimax.SARIMAX
2.
statsmodels.tsa.statespace.sarimax.SARIMAXResults