libcity.evaluator.traffic_state_evaluator

class libcity.evaluator.traffic_state_evaluator.TrafficStateEvaluator(config)[source]

Bases: libcity.evaluator.abstract_evaluator.AbstractEvaluator

clear()[source]

清除之前收集到的 batch 的评估信息，适用于每次评估开始时进行一次清空，排除之前的评估输入的影响。

collect(batch)[source]

收集一 batch 的评估输入

Parameters

batch (dict) – 输入数据，字典类型，包含两个Key:(y_true, y_pred): batch[‘y_true’]: (num_samples/batch_size, timeslots, …, feature_dim) batch[‘y_pred’]: (num_samples/batch_size, timeslots, …, feature_dim)

evaluate()[source]

返回之前收集到的所有 batch 的评估结果

save_result(save_path, filename=None)[source]

将评估结果保存到 save_path 文件夹下的 filename 文件中

Parameters

  • save_path – 保存路径

  • filename – 保存文件名