enerzyme.tasks.metrics.Metrics#

class enerzyme.tasks.metrics.Metrics(metric_config: Dict = {})[source]#

Bases: object

__init__(metric_config: Dict = {}) → None[source]#

cal_judge_score(raw_metric_score: Dict[str, float]) → float[source]#

cal_metric(label: Dict[str, ndarray | List], predict: Dict[str, ndarray | List]) → Dict[str, float][source]#

cal_single_metric(label: Dict[str, ndarray | List], prediction: Dict[str, ndarray | List], target_name: str, metric_name: str) → float[source]#