I love structts model and Kalman filters for forecasting. Sometimes you just want a model that works out of the box without designing a model with a Kalman filter, especially if you need to use long seasonalites and exog variables. Defining all these state space matrices gets tedious pretty quickly...
The code in this repo is an attempt to bring a familiar API to multivariate StructTS model, currently with the simdkalman library as a backend.
pip install simd-structts
- Statsmodels and simdkalman backend implementation.
- Equal filtered/smoothed/predicted states for level/trend models.
- Proper testing for multiple python versions.
- Equal filtered/smoothed/predicted states for exog components.
- Equal filtered/smoothed/predicted states for long seasonal fourier components.
- Passing tests for statsmodels-like initialization of model.
- Pretty API with ABC and stuff.
- Example notebook
- Gradient methods for finding optimal params