wehub-resource-sync
46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
try:
|
|
from typing import Dict
|
|
from typing import List
|
|
from typing import Union
|
|
|
|
from easygraph._global import AUTHOR_EMAIL
|
|
|
|
from .base import BaseEvaluator
|
|
from .classification import VertexClassificationEvaluator
|
|
from .classification import available_classification_metrics
|
|
from .hypergraphs import HypergraphVertexClassificationEvaluator
|
|
except:
|
|
print(
|
|
"Warning raise in module:ml_metrics. Please install Pytorch before you use"
|
|
" functions related to nueral network"
|
|
)
|
|
|
|
|
|
def build_evaluator(
|
|
task: str,
|
|
metric_configs: List[Union[str, Dict[str, dict]]],
|
|
validate_index: int = 0,
|
|
):
|
|
r"""Return the metric evaluator for the given task.
|
|
|
|
Args:
|
|
``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``.
|
|
"""
|
|
if task == "hypergraph_vertex_classification":
|
|
return HypergraphVertexClassificationEvaluator(metric_configs, validate_index)
|
|
else:
|
|
raise ValueError(
|
|
f"{task} is not supported yet. Please email '{AUTHOR_EMAIL}' to add it."
|
|
)
|
|
|
|
|
|
# __all__ = [
|
|
# "BaseEvaluator",
|
|
# "build_evaluator",
|
|
# "available_classification_metrics",
|
|
# "VertexClassificationEvaluator",
|
|
# "HypergraphVertexClassificationEvaluator",
|
|
# ]
|