33 lines
958 B
Markdown
33 lines
958 B
Markdown
依赖包
|
||
支持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还没做好，不要运行。
|