WenjieDu/PyPOTS

View on GitHub
pypots/imputation/timemixer/__init__.py

Summary

Maintainability
A
0 mins
Test Coverage
"""
The package including the modules of TimeMixer.

Refer to the paper
`Shiyu Wang, Haixu Wu, Xiaoming Shi, Tengge Hu, Huakun Luo, Lintao Ma, James Y. Zhang, and Jun Zhou.
"TimeMixer: Decomposable Multiscale Mixing for Time Series Forecasting".
In ICLR 2024.
<https://openreview.net/pdf?id=7oLshfEIC2>`_

Notes
-----
This implementation is inspired by the official one https://github.com/kwuking/TimeMixer

"""

# Created by Wenjie Du <wenjay.du@gmail.com>
# License: BSD-3-Clause


from .model import TimeMixer

__all__ = [
    "TimeMixer",
]