WenjieDu/PyPOTS

View on GitHub
pypots/nn/modules/tide/__init__.py

Summary

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

Refer to the paper
`Abhimanyu Das, Weihao Kong, Andrew Leach, Shaan Mathur, Rajat Sen, and Rose Yu.
"Long-term Forecasting with TiDE: Time-series Dense Encoder".
In Transactions on Machine Learning Research, 2023.
<https://openreview.net/pdf?id=pCbC3aQB5W>`_

Notes
-----
This implementation is inspired by the official one
https://github.com/google-research/google-research/blob/master/tide and https://github.com/lich99/TiDE

"""

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


from .autoencoder import TideEncoder, TideDecoder
from .layers import ResBlock

__all__ = [
    "TideEncoder",
    "TideDecoder",
    "ResBlock",
]