BUAA-DL2021

BUAA-DL2021 copied to clipboard

mjh12138

AI678

GawsonZz

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

除了test目录下的文件，其他文件都经过flake8检查 两个log文件位于Bigscity-TrafficDL目录下

zhangjiahui-buaa

已进行过代码风格检查，由于本模型较复杂（属于困难模型），因此额外添加了相应的`*_evaluator.py`、`*_executor.py`、`*_dataset.py`等文件。 具体修改详情和实验评估细节与结果可参见实验技术报告。

Reinhard-Tichy