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
, anduser_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 to0
.