Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/heidelbergcement/hcrystalball
A library that unifies the API for most commonly used libraries and modeling techniques for time-series forecasting in the Python ecosystem.
https://github.com/heidelbergcement/hcrystalball
cross-validation data-science fbprophet model-selection pmdarima sarimax sklearn sklearn-api sklearn-compatible sklearn-library sktime statsmodels tbats time-series time-series-forecasting transformer wrapper
Last synced: about 1 month ago
JSON representation
A library that unifies the API for most commonly used libraries and modeling techniques for time-series forecasting in the Python ecosystem.
- Host: GitHub
- URL: https://github.com/heidelbergcement/hcrystalball
- Owner: heidelbergcement
- License: mit
- Created: 2020-06-29T11:31:26.000Z (over 4 years ago)
- Default Branch: master
- Last Pushed: 2024-02-21T09:24:08.000Z (9 months ago)
- Last Synced: 2024-09-27T20:22:41.593Z (about 2 months ago)
- Topics: cross-validation, data-science, fbprophet, model-selection, pmdarima, sarimax, sklearn, sklearn-api, sklearn-compatible, sklearn-library, sktime, statsmodels, tbats, time-series, time-series-forecasting, transformer, wrapper
- Language: Python
- Homepage: https://hcrystalball.readthedocs.io/
- Size: 10.9 MB
- Stars: 152
- Watchers: 11
- Forks: 19
- Open Issues: 17
-
Metadata Files:
- Readme: README.md
- Changelog: CHANGELOG.rst
- License: LICENSE.txt
- Code of conduct: CODE_OF_CONDUCT.md
- Authors: AUTHORS.rst
Awesome Lists containing this project
README
[](https://github.com/heidelbergcement/hcrystalball/actions/?query=workflow%3ACI)
[](https://github.com/heidelbergcement/hcrystalball/actions?query=workflow%3ACD)
[](https://hcrystalball.readthedocs.io/en/latest/?badge=latest)
[](https://pypi.org/project/hcrystalball/)
[](https://anaconda.org/conda-forge/hcrystalball)
[](https://codecov.io/gh/heidelbergcement/hcrystalball)
[](https://mybinder.org/v2/gh/heidelbergcement/hcrystalball/master?filepath=docs/examples/)
[](https://raw.githubusercontent.com/heidelbergcement/hcrystalball/master/LICENSE.txt)
[](https://github.com/heidelbergcement/hcrystalball/graphs/contributors)
[](https://github.com/psf/black)
# HCrystal Ball
A library that unifies the API for most commonly
used libraries and modelling techniques for time-series
forecasting in the Python ecosystem.
**HCrystal Ball** consists of two main parts:
* **Wrappers** - which bring different 3rd party
libraries to time series compatible sklearn API
* **Model Selection** - to enable gridsearch over wrappers, general or custom made transformers
and add convenient layer over whole process (access to results, plots, storage, ...)
## Documentation
See examples, tutorials, contribution, API and more on the documentation [site](https://hcrystalball.readthedocs.io/en/latest) try notebooks on [binder](https://mybinder.org/v2/gh/heidelbergcement/hcrystalball/master) or browse example notebooks in [docs/examples](https://github.com/heidelbergcement/hcrystalball/tree/master/docs/examples) directly.
## Core Installation
If you want really minimal installation, you can install from pip or from conda-forge
```bash
pip install hcrystalball
```
```bash
conda install -c conda-forge hcrystalball
```
## Typical Installation
Very often you will want to use more wrappers, than just Sklearn, run examples in jupyterlab, or execute model selection in parallel. Getting such dependencies to play together nicely might be cumbersome, so checking `envrionment.yml` might give you faster start.
```bash
# get dependencies file, e.g. using curl
curl -O https://raw.githubusercontent.com/heidelbergcement/hcrystalball/master/environment.yml
# check comments in environment.yml, keep or remove as requested, than create environment using
conda env create -f environment.yml
# activate the environment
conda activate hcrystalball
# if you want to see progress bar in jupyterlab, execute also
jupyter labextension install @jupyter-widgets/jupyterlab-manager
# install the library from pip
pip install hcrystalball
# or from conda
conda install -c conda-forge hcrystalball
```
## Development Installation:
To have everything in place including docs build or executing tests, execute following code
```bash
git clone https://github.com/heidelbergcement/hcrystalball
cd hcrystalball
conda env create -f environment.yml
conda activate hcrystalball
# ensures interactive progress bar will work in example notebooks
jupyter labextension install @jupyter-widgets/jupyterlab-manager
python setup.py develop
```
## Example Usage
### Wrappers
```python
from hcrystalball.utils import generate_tsdata
from hcrystalball.wrappers import ProphetWrapper
X, y = generate_tsdata(n_dates=365*2)
X_train, y_train, X_test, y_test = X[:-10], y[:-10], X[-10:], y[-10:]
model = ProphetWrapper()
y_pred = model.fit(X_train, y_train).predict(X_test)
y_pred
prophet
2018-12-22 6.066999
2018-12-23 6.050076
2018-12-24 6.105620
2018-12-25 6.141953
2018-12-26 6.150229
2018-12-27 6.163615
2018-12-28 6.147420
2018-12-29 6.048633
2018-12-30 6.031711
2018-12-31 6.087255
```
### Model Selection
```python
import pandas as pd
import matplotlib.pyplot as plt
plt.style.use('seaborn')
plt.rcParams['figure.figsize'] = [12, 6]
from hcrystalball.utils import get_sales_data
from hcrystalball.model_selection import ModelSelector
df = get_sales_data(n_dates=200,
n_assortments=1,
n_states=2,
n_stores=2)
ms = ModelSelector(horizon=10,
frequency="D",
country_code_column="HolidayCode",
)
ms.create_gridsearch(n_splits=2,
sklearn_models=True,
prophet_models=False,
exog_cols=["Open","Promo","SchoolHoliday","Promo2"],
)
ms.select_model(df=df,
target_col_name="Sales",
partition_columns=["Assortment", "State","Store"],
)
ms.plot_results(plot_from="2015-06-01",
partitions=[{"Assortment":"a","State":"NW","Store":335}]
)
```
