|
|
||
|---|---|---|
| assets | ||
| config | ||
| dataloader | ||
| lib | ||
| model | ||
| trainer | ||
| .gitignore | ||
| LICENSE | ||
| README.md | ||
| baseline.ipynb | ||
| run.py | ||
README.md
依赖包
pip install pyyaml tqdm statsmodels h5py pip3 install torch torchvision torchaudio
命令
添加模型:
三步:1. 在config下新建文件夹,复制其他模型的参数项,改命令
- 在model新建文件夹,复制模型文件
使用 arg['参数名']访问参数,具体参数在yaml文件中的model类下对应配置,有啥写啥。一般只要这里的参数就可以了,不需要动其他的train,data
第三步:在model/model_selector下添加自己的模型,类似这种格式
然后就可以运行了。其中,ARIMA,VAR还没做好,不要运行。



