Metadata-Version: 2.1
Name: sklearn-ts
Version: 0.0.5
Summary: Package for time series forecasting
Home-page: https://github.com/lady-pandas/sklearn-ts
Author: Marta Markiewicz
Author-email: m.markiewicz.pl@gmail.com
License: MIT
Keywords: time series forecasting deep learning machine learning regression ARIMA ETS
Platform: UNKNOWN
Description-Content-Type: text/markdown

# Welcome to sklearn-ts

Testing time series forecasting models made easy :)
This package leverages [scikit-learn](https://github.com/scikit-learn/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:
- [sktime](https://github.com/alan-turing-institute/sktime)
- [sktime-dl](https://github.com/sktime/sktime-dl)
- [darts](https://github.com/unit8co/darts)

# Installation

```bash
pip install sklearn-ts
```

# Quickstart
## Forecasting COVID-19 with Linear Regression
```python
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](tests\LinearRegression.png)

# Forecasting models

| Model family | Model | Univariate |
| ------------- |:-------------:| -----:|
| Benchmark | Naive | 1 |
| Exponential Smoothing | SES | 1 |
| Exponential Smoothing | Holt's linear | 1 |
| Exponential Smoothing | Holt-Winter | 1 |
| - | Prophet |  |
| Neural networks | ANN |  |
| Neural networks | LSTM |  |
| Neural networks | TCN |  |

# Documentation
Tutorial notebooks:
- [neural networks](https://colab.research.google.com/drive/1wSZPydSkIoGYh9VANgP_wTQe-wrhzY1w#scrollTo=_W2QP0dhCKFV)

# Development roadmap
- Memes
- wtorek - czy działa
- FEATURES + SHAP
- x13
- prettier plot
- Alphabet quiz
- Handling many observations per date
- Constant window for forecasting
- For NN - chart of how it learned
- Logging
- Read the docs
- prod
- save picture optional
- PI Coverage
- Watermark


