libcity.evaluator.traffic_accident_evaluator¶

class libcity.evaluator.traffic_accident_evaluator.TrafficAccidentEvaluator(config)[源代码]¶

基类：libcity.evaluator.traffic_state_evaluator.TrafficStateEvaluator

collect(batch)[源代码]¶

收集一 batch 的评估输入

参数: 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)