libcity.executor.chebconv_executor¶

class libcity.executor.chebconv_executor.ChebConvExecutor(config, model, data_feature)[source]¶

Bases: libcity.executor.traffic_state_executor.TrafficStateExecutor

_train_epoch(train_dataloader, epoch_idx, loss_func=None)[source]¶

完成模型一个轮次的训练

Returns: 训练集的损失值
Return type: float

_valid_epoch(eval_dataloader, epoch_idx, loss_func=None)[source]¶

完成模型一个轮次的评估

Parameters

eval_dataloader – 评估数据
epoch_idx – 轮次数
loss_func – 损失函数

Returns

验证集的损失值

Return type

float

evaluate(test_dataloader)[source]¶: use model to test data

train(train_dataloader, eval_dataloader)[source]¶: use data to train model with config