easygraph.ml_metrics package

Subpackages

• easygraph.ml_metrics.hypergraphs package
  • Submodules
    • easygraph.ml_metrics.hypergraphs.hypergraph module
      • HypergraphVertexClassificationEvaluator
  • Module contents

Submodules

• easygraph.ml_metrics.base module
  • BaseEvaluator
    • BaseEvaluator.test()
    • BaseEvaluator.test_add_batch()
    • BaseEvaluator.test_epoch_res()
    • BaseEvaluator.validate()
    • BaseEvaluator.validate_add_batch()
    • BaseEvaluator.validate_epoch_res()
  • format_metric_configs()
• easygraph.ml_metrics.classification module
  • VertexClassificationEvaluator
    • VertexClassificationEvaluator.test()
    • VertexClassificationEvaluator.validate()
  • accuracy()
  • available_classification_metrics()
  • confusion_matrix()
  • f1_score()

Module contents

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

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.