pypots/nn/modules/fedformer/__init__.py
"""
The package including the modules of FEDformer.
Refer to the paper
`Tian Zhou, Ziqing Ma, Qingsong Wen, Xue Wang, Liang Sun, and Rong Jin.
FEDformer: Frequency enhanced decomposed transformer for long-term series forecasting.
In ICML, volume 162 of Proceedings of Machine Learning Research, pages 27268–27286. PMLR, 17–23 Jul 2022.
<https://proceedings.mlr.press/v162/zhou22g/zhou22g.pdf>`_
Notes
-----
This implementation is inspired by the official one https://github.com/MAZiqing/FEDformer
"""
# Created by Wenjie Du <wenjay.du@gmail.com>
# License: BSD-3-Clause
from .autoencoder import FEDformerEncoder, FEDformerDecoder
from .layers import (
MultiWaveletTransform,
MultiWaveletCross,
FourierBlock,
FourierCrossAttention,
)
__all__ = [
"FEDformerEncoder",
"FEDformerDecoder",
"MultiWaveletTransform",
"MultiWaveletCross",
"FourierBlock",
"FourierCrossAttention",
]