From dc69f08f17c32cc59fe566f86b9856e016179794 Mon Sep 17 00:00:00 2001 From: arminwitte <110226001+arminwitte@users.noreply.github.com> Date: Sun, 24 Sep 2023 10:06:11 +0200 Subject: [PATCH] Corrected copy paste error where regularisation metric is never chosen by factory --- binarybeech/metrics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)