BUAA-DL2021
BUAA-DL2021 copied to clipboard
ATST-LSTM
- 修改了Bigscity-TrafficDL/test_model.py:从而能够测试新的模型
- 修改了Bigscity-TrafficDL/run_model.py:从而能够训练并且评估新的模型
- 增加了Bigscity-TrafficDL/trafficdl/data/dataset/atst_lstm_dataset.py:作为ATSTLSTM的数据处理类
- 增加了Bigscity-TrafficDL/trafficdl/model/trajectory_loc_prediction/ATSTLSTM.py:ATSTLSTM模型的实现
- 增加了一个config文件:Bigscity-TrafficDL/trafficdl/config/data/ATSTLSTMDataset.json
- 增加了一个config文件:Bigscity-TrafficDL/trafficdl/config/model/ATSTLSTM.json
- 修改了Bigscity-TrafficDL/trafficdl/config/task_config.json:其中加入了ATSTLSTM和相关参数
- 修改了Bigscity-TrafficDL/trafficdl/data/dataset/init.py:其中加入了ATSTLSTMDataset
- 修改了Bigscity-TrafficDL/trafficdl/model/init.py:其中加入了ATSTLSTM
- 修改了Bigscity-TrafficDL/trafficdl/model/trajectory_loc_prediction/init.py:其中加入了ATSTLSTM