交通轮,基线合集
Go to file
HengZhang be5e810c54 mv dir name 2025-03-03 10:29:42 +08:00
assets init 2025-03-02 23:41:12 +08:00
config init 2025-03-02 23:41:12 +08:00
dataloader mv dir name 2025-03-03 10:29:42 +08:00
lib init 2025-03-02 23:41:12 +08:00
model mv dir name 2025-03-03 10:29:42 +08:00
trainer init 2025-03-02 23:41:12 +08:00
.gitignore mv dir name 2025-03-03 10:29:42 +08:00
LICENSE Initial commit 2025-03-02 23:39:13 +08:00
README.md init 2025-03-02 23:41:12 +08:00
baseline.ipynb init 2025-03-02 23:41:12 +08:00
run.py init 2025-03-02 23:41:12 +08:00

README.md

依赖包

pip install pyyaml tqdm statsmodels h5py pip3 install torch torchvision torchaudio

命令

image-20241214230153502

添加模型:

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

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

image-20241214230303239

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

image-20241214230331273

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

image-20241214230447678

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