libcity笔记:添加新模型(以RNN.py为例)
- 创建的新模型应该继承
AbstractModel
或AbstractTrafficStateModel
- 交通状态预测任务——>继承
AbstractTrafficStateModel
类 - 轨迹位置预测任务——>继承
AbstractModel
类
- 交通状态预测任务——>继承
1 AbstractTrafficStateModel
2 RNN
2.1 构造函数
2.2 predict