easygraph.ml_metrics package

Subpackages

Submodules

easygraph.ml_metrics.base module

easygraph.ml_metrics.classification module

Module contents

easygraph.ml_metrics.build_evaluator(task: str, metric_configs: List[Union[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.