This project has been quarantined by PyPI admins. It cannot be installed or modified until a security review is complete.
ETNA Time Series Library
Predict your time series the easiest way
Homepage | Documentation | Tutorials | Contribution Guide | Release Notes
ETNA is an easy-to-use time series forecasting framework. It includes built in toolkits for time series preprocessing, feature generation, a variety of predictive models with unified interface - from classic machine learning to SOTA neural networks, models combination methods and smart backtesting. ETNA is designed to make working with time series simple, productive, and fun.
ETNA is the first python open source framework of Tinkoff.ru Artificial Intelligence Center. The library started as an internal product in our company - we use it in over 10+ projects now, so we often release updates. Contributions are welcome - check our Contribution Guide.
Installation
ETNA is on PyPI, so you can use pip to install it.
pip install --upgrade pip
pip install etna
Get started
Here's some example code for a quick start.
import pandas as pd
from etna.datasets.tsdataset import TSDataset
from etna.models import ProphetModel
from etna.pipeline import Pipeline
# Read the data
df = pd.read_csv("examples/data/example_dataset.csv")
# Create a TSDataset
df = TSDataset.to_dataset(df)
ts = TSDataset(df, freq="D")
# Choose a horizon
HORIZON = 8
# Fit the pipeline
pipeline = Pipeline(model=ProphetModel(), horizon=HORIZON)
pipeline.fit(ts)
# Make the forecast
forecast_ts = pipeline.forecast()
Tutorials
We have also prepared a set of tutorials for an easy introduction:
| Notebook | Interactive launch |
|---|---|
| Get started | |
| Backtest | |
| EDA | |
| Outliers | |
| Clustering | |
| Deep learning models | |
| Ensembles |
Documentation
ETNA documentation is available here.
Resources
-
Forecasting with ETNA: Fast and Furious on Medium
-
Store sales prediction with etna library on Kaggle
-
PyCon Russia September 2021 talk on YouTube
-
ETNA Tutorial. Part 1 (Rus), ETNA Tutorial. Part 2 (Rus) on YouTube thanks to RC Gewissta
Acknowledgments
ETNA.Team
Andrey Alekseev, Nikita Barinov, Dmitriy Bunin, Aleksandr Chikov, Vladislav Denisov, Martin Gabdushev, Sergey Kolesnikov, Artem Makhin, Ivan Mitskovets, Albina Munirova, Nikolay Romantsov, Julia Shenshina
ETNA.Contributors
Artem Levashov, Aleksey Podkidyshev, Carlosbogo
License
Feel free to use our library in your commercial and private applications.
ETNA is covered by Apache 2.0. Read more about this license here
Metadata
Release files for etna 1.6.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| etna-1.6.0.tar.gz | 115.0 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| etna-1.6.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 281.2 kB
Release files / etna-1.6.0.tar.gz
| Download URL | etna-1.6.0.tar.gz |
|---|---|
| Size | 115.0 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
5f37b17831a233f07ffa822901a131633d9243346594dfae66a81b1780e80e84
|
|
BLAKE2b-256 checksum How to use checksums |
b31afa50557c349913c619d89e841ecc65c6056945bdea53cf6022857eb2c375
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
poetry/1.1.12 CPython/3.9.10 Linux/5.11.0-1027-azure
|
Release files / etna-1.6.0-py3-none-any.whl
| Download URL | etna-1.6.0-py3-none-any.whl |
|---|---|
| Size | 166.2 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
69fc1334b68c660f959458e71a48a26422d33d218f45e82dd309cd0555096768
|
|
BLAKE2b-256 checksum How to use checksums |
4f2d5091ba60c5f1c9644598da657903aa271fda3d45e6efd2d315cc70c20221
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
poetry/1.1.12 CPython/3.9.10 Linux/5.11.0-1027-azure
|