libcity.model.abstract_traffic_state_model¶
-
class
libcity.model.abstract_traffic_state_model.AbstractTrafficStateModel(config, data_feature)[source]¶ Bases:
libcity.model.abstract_model.AbstractModel-
calculate_loss(batch)[source]¶ 输入一个batch的数据，返回训练过程的loss，也就是需要定义一个loss函数
- Parameters
batch (Batch) – a batch of input
- Returns
return training loss
- Return type
torch.tensor
-
predict(batch)[source]¶ 输入一个batch的数据，返回对应的预测值，一般应该是**多步预测**的结果，一般会调用nn.Moudle的forward()方法
- Parameters
batch (Batch) – a batch of input
- Returns
predict result of this batch
- Return type
torch.tensor
-
training: bool¶
-