diff --git a/binarybeech/metrics.py b/binarybeech/metrics.py index 4a8fd1e..9ab80de 100644 --- a/binarybeech/metrics.py +++ b/binarybeech/metrics.py @@ -398,7 +398,7 @@ def from_data(self, y, algorithm_kwargs): metrics_factory = MetricsFactory() metrics_factory.register("regression", RegressionMetrics) -metrics_factory.register("regression:regularized", RegressionMetrics) +metrics_factory.register("regression:regularized", RegressionMetricsRegularized) metrics_factory.register("classification:gini", ClassificationMetrics) metrics_factory.register("classification:entropy", ClassificationMetricsEntropy) metrics_factory.register("logistic", LogisticMetrics)