package.experimental.train.metrics.registry.MetricsRegistry Class Reference

Static Public Member Functions
Metric	create_metric (str name)

Static Protected Attributes
dict	_metrics

Detailed Description

Registry for creating metrics by name.

Definition at line 6 of file registry.py.

Member Function Documentation

create_metric()

Metric package.experimental.train.metrics.registry.MetricsRegistry.create_metric ( str  name )

static

Create a metric instance by name.

Args:
    name (str): Name of the metric.

Returns:
    Metric: Metric instance.

Definition at line 15 of file registry.py.

    def create_metric(name: str) -> Metric:
        """
        Create a metric instance by name.
 
        Args:
            name (str): Name of the metric.
 
        Returns:
            Metric: Metric instance.
        """
        if name not in MetricsRegistry._metrics:
            raise ValueError(
                f"Unknown Metric: {name}. Custom metric is not supported yet")
        return MetricsRegistry._metrics[name]()

Field Documentation

_metrics

dict package.experimental.train.metrics.registry.MetricsRegistry._metrics

staticprotected

Initial value:

=  {
        "categorical_accuracy": CategoricalAccuracy,
    }

Definition at line 10 of file registry.py.

---

The documentation for this class was generated from the following file:

• runtime/onert/api/python/package/experimental/train/metrics/registry.py