TrafficWheel/README.md

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

conda create -n trafficwheel python=3.10

pip install pyyaml tqdm statsmodels h5py kagglehub torch torchvision torchaudio torchdiffeq fastdtw

命令

添加模型：

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

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

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

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

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