BUAA-DL2021 icon indicating copy to clipboard operation
BUAA-DL2021 copied to clipboard

ATST-LSTM

Open CaiTH0618 opened this issue 3 years ago • 0 comments

  • 修改了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

CaiTH0618 avatar Apr 20 '21 06:04 CaiTH0618