sklearn-ts

Package for time series forecasting


Keywords
time, series, forecasting, deep, learning, machine, regression, ARIMA, ETS
License
MIT
Install
pip install sklearn-ts==0.0.6

Documentation

Welcome to sklearn-ts

Testing time series forecasting models made easy :) This package leverages scikit-learn, simply tuning it where needed for time series specific purposes.

Main features include:

  • Moving window time split
    • train-test split
    • CV on moving window time splits
  • Model wrappers:
    • Neural networks

Other python packages in the time series domain:

Installation

pip install sklearn-ts

Quickstart

Forecasting COVID-19 with Linear Regression

from sklearn_ts.datasets.covid import load_covid
from sklearn.linear_model import LinearRegression
from sklearn_ts.validator import check_model

dataset = load_covid()['dataset']
dataset['month'] = dataset['date'].dt.month

params = {'fit_intercept': [False, True]}
regressor = LinearRegression()

results = check_model(
    regressor, params, dataset,
    target='new_cases', features=['month'], categorical_features=[], user_transformers=[],
    h=14, n_splits=2, gap=14,
    plotting=True
)

alt text

Forecasting models

Model family Model Univariate
Neural networks ANN
Neural networks LSTM
Neural networks TCN

Documentation

Tutorial notebooks:

Development roadmap

  • usunąć printy
  • instrukcja
  • prerequisites
  • Outliers
  • Transform plots
  • High level summary
  • Handling many observations per date
  • Constant window for forecasting
  • Image not included
  • For NN - chart of how it learned
  • Logging
  • Read the docs
  • CRISP
  • SHAP
  • prod
  • save picture optioal
  • not only mae
  • single or collective
  • memes
  • area plot
  • feature importances
  • watermark
  • spójnie - intuicja, wzory
  • ARIMA odpowiedniki e Exp