easygraph.ml_metrics package#

Subpackages#

Submodules#

Module contents#

easygraph.ml_metrics.build_evaluator(task: str, metric_configs: List[str | Dict[str, dict]], validate_index: int = 0)[source]#

Return the metric evaluator for the given task.

Parameters:
  • task (str) – The type of the task. The supported types include: graph_vertex_classification, hypergraph_vertex_classification, and user_item_recommender.

  • metric_configs (List[Union[str, Dict[str, dict]]]) – The list of metric names.

  • validate_index (int) – The specified metric index used for validation. Defaults to 0.