Loading repository data…
Loading repository data…
facebookresearch / repository
Kats, a kit to analyze time series data, a lightweight, easy-to-use, generalizable, and extendable framework to perform time series analysis, from understanding the key statistics and characteristics, detecting change points and anomalies, to forecasting future trends.
A transparent discovery signal based on current public GitHub metadata.
This score does not audit code, security, maintainers, documentation quality, or suitability. Verify the repository and its current documentation before adoption.
Kats is a toolkit to analyze time series data, a lightweight, easy-to-use, and generalizable framework to perform time series analysis. Time series analysis is an essential component of Data Science and Engineering work at industry, from understanding the key statistics and characteristics, detecting regressions and anomalies, to forecasting future trends. Kats aims to provide the one-stop shop for time series analysis, including detection, forecasting, feature extraction/embedding, multivariate analysis, etc.
Kats is released by Facebook's Infrastructure Data Science team. It is available for download on PyPI.
Kats is on PyPI, so you can use pip to install it.
pip install --upgrade pip
pip install kats
If you need only a small subset of Kats, you can install a minimal version of Kats with
MINIMAL_KATS=1 pip install kats
which omits many dependencies (everything in test_requirements.txt).
However, this will disable many functionalities and cause import kats to log
warnings. See setup.py for full details and options.
Here are a few sample snippets from a subset of Kats offerings:
Using Prophet model to forecast the air_passengers data set.
import pandas as pd
from kats.consts import TimeSeriesData
from kats.models.prophet import ProphetModel, ProphetParams
# take `air_passengers` data as an example
air_passengers_df = pd.read_csv(
"../kats/data/air_passengers.csv",
header=0,
names=["time", "passengers"],
)
# convert to TimeSeriesData object
air_passengers_ts = TimeSeriesData(air_passengers_df)
# create a model param instance
params = ProphetParams(seasonality_mode='multiplicative') # additive mode gives worse results
# create a prophet model instance
m = ProphetModel(air_passengers_ts, params)
# fit model simply by calling m.fit()
m.fit()
# make prediction for next 30 month
fcst = m.predict(steps=30, freq="MS")
Using CUSUM detection algorithm on simulated data set.
# import packages
import numpy as np
import pandas as pd
from kats.consts import TimeSeriesData
from kats.detectors.cusum_detection import CUSUMDetector
# simulate time series with increase
np.random.seed(10)
df_increase = pd.DataFrame(
{
'time': pd.date_range('2019-01-01', '2019-03-01'),
'increase':np.concatenate([np.random.normal(1,0.2,30), np.random.normal(2,0.2,30)]),
}
)
# convert to TimeSeriesData object
timeseries = TimeSeriesData(df_increase)
# run detector and find change points
change_points = CUSUMDetector(timeseries).detector()
We can extract meaningful features from the given time series data
# Initiate feature extraction class
import pandas as pd
from kats.consts import TimeSeriesData
from kats.tsfeatures.tsfeatures import TsFeatures
# take `air_passengers` data as an example
air_passengers_df = pd.read_csv(
"../kats/data/air_passengers.csv",
header=0,
names=["time", "passengers"],
)
# convert to TimeSeriesData object
air_passengers_ts = TimeSeriesData(air_passengers_df)
# calculate the TsFeatures
features = TsFeatures().transform(air_passengers_ts)
If you use Kats in your work or research, please use the following BibTeX entry.
@software{Jiang_KATS_2022,
author = {Jiang, Xiaodong and Srivastava, Sudeep and Chatterjee, Sourav and Yu, Yang and Handler, Jeffrey and Zhang, Peiyi and Bopardikar, Rohan and Li, Dawei and Lin, Yanjun and Thakore, Uttam and Brundage, Michael and Holt, Ginger and Komurlu, Caner and Nagalla, Rakshita and Wang, Zhichao and Sun, Hechao and Gao, Peng and Cheung, Wei and Gao, Jun and Wang, Qi and Guerard, Marius and Kazemi, Morteza and Chen, Yulin and Zhou, Chong and Lee, Sean and Laptev, Nikolay and Levendovszky, Tihamér and Taylor, Jake and Qian, Huijun and Zhang, Jian and Shoydokova, Aida and Singh, Trisha and Zhu, Chengjun and Baz, Zeynep and Bergmeir, Christoph and Yu, Di and Koylan, Ahmet and Jiang, Kun and Temiyasathit, Ploy and Yurtbay, Emre},
license = {MIT License},
month = {3},
title = {{Kats}},
url = {https://github.com/facebookresearch/Kats},
version = {0.2.0},
year = {2022}
}
Core Algorithm
A self-supervised framework designed to automatically optimize hyper-parameters for time-series forecasting models at scale. The method reduces tuning cost, improves generalization across heterogeneous datasets, and supports efficient deployment in large production forecasting environments. The accompanying research was presented as an invited talk at international forecasting venues and has been referenced in work on automated model selection and adaptive time-series learning.
Meta AI Official post: Large-scale forecasting: Self-supervised learning framework for hyperparameter tuning
Paper: Self-supervised Learning for Fast and Scalable Time Series Hyper-parameter Tuning
Invited talk: the 40th International Symposium on Forecasting.
If you use this algorithm in your work or research, please use the following BibTeX entry.
@article{Zhang_SSL-HPT_2021,
author = {Zhang, Peiyi and Jiang, Xiaodong and Holt, Ginger M. and Laptev, Nikolay Pavlovich and Komurlu, Caner and Gao, Peng and Yu, Yang},
title = {Self-supervised Learning for Fast and Scalable Time Series Hyper-parameter Tuning},
year = {2021},
eprint = {2102.05740},
archivePrefix = {arXiv},
primaryClass = {cs.LG},
url = {https://arxiv.org/pdf/2102.05740}
}
Kats is currentely maintaned by community with the main contributions and leading from Nickolai Kniazev and Peter Shaffery
Kats is a project with several skillful researchers and engineers contributing to it. Kats was started and built by Xiaodong Jiang with major contributions coming from many talented individuals in various forms and means. A non-exhaustive but growing list needs to mention: Sudeep Srivastava, Sourav Chatterjee, Jeff Handler, Rohan Bopardikar, Dawei Li, Yanjun Lin, Yang Yu, Michael Brundage, Caner Komurlu, Rakshita Nagalla, Zhichao Wang, Hechao Sun, Peng Gao, Wei Cheung, Jun Gao, Qi Wang, Morteza Kazemi, Tihamér Levendovszky, Jian Zhang, Ahmet Koylan, Kun Jiang, Aida Shoydokova, Ploy Temiyasathit, Sean Lee, , , , , , , , , , , , , , , and .
Kats is licensed under the MIT license.