KAN4TSF is an official PyTorch implementation of KAN4TSF: Are KAN and KAN-based models Effective for Time Series Forecasting? Although it is called KAN4TSF, it also supports time series forecasting methods of various network structures (CNN, Linear, Transformer and others).
🚩 News (2024.09) Introduction and Reproduction (in Chinese)
Step by Step with Conda:
conda create -n kan4tsf python=3.10
conda activate kan4tsf
conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia
python -m pip install lightningor you can just:
pip install -r requirements.txtETTh1 and ETTm1 can be downloaded within this project, and other datasets can be downloaded from [Baidu Drive].
python train.py -c config/reproduce_conf/RMoK/ETTh1_96for96.pyIf you find this repo useful, please cite our paper:
@inproceedings{han2023are,
title={Are KANs Effective for Time Series Forecasting?},
author={Xiao Han, Xinfeng Zhang, Yiling Wu, Zhenduo Zhang and Zhe Wu},
booktitle={arXiv},
year={2024},
}