依赖包
支持python 3.10以上版本。

conda create -n trafficwheel python=3.10
pip install pyyaml tqdm statsmodels h5py kagglehub torch torchvision torchaudio torchdiffeq fastdtw



命令

![image-20241214230153502](./assets/image-20241214230153502.png)



添加模型：

三步：1. 在config下新建文件夹，复制其他模型的参数项，改命令

2. 在model新建文件夹，复制模型文件

![image-20241214230303239](./assets/image-20241214230303239.png)

使用 arg['参数名']访问参数，具体参数在yaml文件中的model类下对应配置，有啥写啥。一般只要这里的参数就可以了，不需要动其他的train,data

![image-20241214230331273](./assets/image-20241214230331273.png)

第三步：在model/model_selector下添加自己的模型，类似这种格式

![image-20241214230447678](./assets/image-20241214230447678.png)

然后就可以运行了。其中，ARIMA,VAR还没做好，不要运行。