libcity笔记：添加新模型（以RNN.py为例）

本文主要是介绍libcity笔记：添加新模型（以RNN.py为例），希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

• 创建的新模型应该继承AbstractModel或AbstractTrafficStateModel
  • 交通状态预测任务——>继承 AbstractTrafficStateModel类
  • 轨迹位置预测任务——>继承AbstractModel类

1 AbstractTrafficStateModel

2 RNN

2.1 构造函数

2.2 predict

 

2.3 calculate_loss

这篇关于libcity笔记：添加新模型（以RNN.py为例）的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---