SkillAgentSearch skills...

Tslearn

The machine learning toolkit for time series analysis in Python

Install / Use

/learn @tslearn-team/Tslearn

README

<!-- Our logo and description --> <div align="center"> <p><a href="https://tslearn.readthedocs.io"><image src="https://raw.githubusercontent.com/tslearn-team/tslearn/main/docs/_static/tslearn_logo_white_background.png?cache-control=no-cache" width="20%" alt="tslearn logo"/></a></p> <h1>The machine learning toolkit for time series analysis in Python</h1> </div> <!-- The badges --> <p align="center"> <a href="https://badge.fury.io/py/tslearn"> <img alt="PyPI" src="https://badge.fury.io/py/tslearn.svg?cache-control=no-cache"> </a> <a href="https://www.python.org/downloads/"> <img alt="Python 3.10+" src="https://img.shields.io/badge/python-3.10+-blue.svg"> </a> <a href="http://tslearn.readthedocs.io/en/stable/?badge=stable"> <img alt="Documentation" src="https://readthedocs.org/projects/tslearn/badge/?version=stable"> </a> <a href="https://dev.azure.com/romaintavenard/tslearn/_build"> <img alt="Build (Azure Pipelines)" src="https://dev.azure.com/romaintavenard/tslearn/_apis/build/status/tslearn-team.tslearn?branchName=main"> </a> <a href="https://codecov.io/gh/tslearn-team/tslearn"> <img alt="Codecov" src="https://codecov.io/gh/tslearn-team/tslearn/branch/main/graph/badge.svg"> </a> <a href="https://pepy.tech/project/tslearn"> <img alt="Downloads" src="https://pepy.tech/badge/tslearn"> </a> </p> <!-- Draw horizontal rule --> <hr> <!-- Table of content -->

| Section | Description | |-|-| | Installation | Installing the dependencies and tslearn | | Getting started | A quick introduction on how to use tslearn | | Available features | An extensive overview of tslearn's functionalities | | Documentation | A link to our API reference and a gallery of examples | | Contributing | A guide for heroes willing to contribute | | Citation | A citation for tslearn for scholarly articles |

Installation<a id="installation"></a>

There are different alternatives to install tslearn:

  • PyPi: python -m pip install tslearn
  • Conda: conda install -c conda-forge tslearn
  • Git: python -m pip install https://github.com/tslearn-team/tslearn/archive/main.zip

In order for the installation to be successful, the required dependencies must be installed. For a more detailed guide on how to install tslearn, please see the Documentation.

Getting started<a id="getting-started"></a>

1. Getting the data in the right format

tslearn expects a time series dataset to be formatted as a 3D numpy array. The three dimensions correspond to the number of time series, the number of measurements per time series and the number of dimensions respectively (n_ts, max_sz, d). In order to get the data in the right format, different solutions exist:

It should further be noted that tslearn supports variable-length timeseries.

>>> from tslearn.utils import to_time_series_dataset
>>> my_first_time_series = [1, 3, 4, 2]
>>> my_second_time_series = [1, 2, 4, 2]
>>> my_third_time_series = [1, 2, 4, 2, 2]
>>> X = to_time_series_dataset([my_first_time_series,
                                my_second_time_series,
                                my_third_time_series])
>>> y = [0, 1, 1]

2. Data preprocessing and transformations

Optionally, tslearn has several utilities to preprocess the data. In order to facilitate the convergence of different algorithms, you can scale time series. Alternatively, in order to speed up training times, one can resample the data or apply a piece-wise transformation.

>>> from tslearn.preprocessing import TimeSeriesScalerMinMax
>>> X_scaled = TimeSeriesScalerMinMax().fit_transform(X)
>>> print(X_scaled)
[[[0.] [0.667] [1.] [0.333] [nan]]
 [[0.] [0.333] [1.] [0.333] [nan]]
 [[0.] [0.333] [1.] [0.333] [0.333]]]

3. Training a model

After getting the data in the right format, a model can be trained. Depending on the use case, tslearn supports different tasks: classification, clustering and regression. For an extensive overview of possibilities, check out our gallery of examples.

>>> from tslearn.neighbors import KNeighborsTimeSeriesClassifier
>>> knn = KNeighborsTimeSeriesClassifier(n_neighbors=1)
>>> knn.fit(X_scaled, y)
>>> print(knn.predict(X_scaled))
[0 1 1]

As can be seen, the models in tslearn follow the same API as those of the well-known scikit-learn. Moreover, they are fully compatible with it, allowing to use different scikit-learn utilities such as hyper-parameter tuning and pipelines.

4. More analyses

tslearn further allows to perform all different types of analysis. Examples include calculating barycenters of a group of time series or calculate the distances between time series using a variety of distance metrics.

Available features<a id="available-features"></a>

| data | processing | clustering | classification | regression | metrics | |----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|-------------------------------------------------------------------------------------------------------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------|-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|--------------------------------------------------------------------------------------------------------------------------------------| | UCR Datasets | Scaling | TimeSeriesKMeans | KNN Classifier | KNN Regressor | Dynamic Time Warping | | Generators | Piecewise | KShape | [TimeSeriesSVC](https://tslearn.readthedocs.io/en/stable/gen_modules/svm/tslearn.svm.TimeSeriesSVC.html#tslearn.svm.

Related Skills

View on GitHub
GitHub Stars3.1k
CategoryEducation
Updated58m ago
Forks368

Languages

Python

Security Score

100/100

Audited on Mar 28, 2026

No findings