l5kit.cle.scene_type_agg module

l5kit.cle.scene_type_agg.compute_cle_scene_type_aggregations(mset: l5kit.cle.metric_set.L5MetricSet, scene_ids_to_scene_types: List[List[str]], list_validator_table_to_publish: List[str]) → Dict[str, torch.Tensor]

Compute the scene-type metric aggregations.

Parameters

• mset – metric set to aggregate by scene type

• scene_ids_to_scene_types – list of scene type tags per scene

• list_validator_table_to_publish – list of validators for which we return structured dictionary of results

Returns

dict of result key “scene_type/validator_name” to scale tensor aggregation value.