nntimeseries
该存储库提供了论文的代码,以及在keras模型上运行网格serach的通用代码。
文件'nnts / models / {CNN,LSTM,LSTM2,LR,SOCNN} .py'提供了用于测试各个模型的代码,最后一个实现了建议的重要性偏移CNN,而LSTM2实现了多层LSTM。
基本用法
每个模型文件都可以作为脚本运行,例如
python ./CNN.py --dataset=artificial #默认保存文件
python ./SOCNN.py --dataset=household --save_file=results/household_0.pkl
可以在上述每个文件中指定用于网格搜索的参数。
每个文件都定义了一个模型类,可以将其导入并在外部数据集上使用,如example.ipynb文件所示。
该存储库支持在人工多元嘈杂AR时间序列和家庭用
1