From 2bc5e81a5587333d3f9608ae0dbd98a3601db858 Mon Sep 17 00:00:00 2001
From: Dominique <dominique.benielli@univ-amu.fr>
Date: Wed, 26 Feb 2025 17:03:37 +0100
Subject: [PATCH] Update base.py
---
summit/multiview_platform/utils/base.py | 14 +++++++-------
1 file changed, 7 insertions(+), 7 deletions(-)
diff --git a/summit/multiview_platform/utils/base.py b/summit/multiview_platform/utils/base.py
index 67df47a6..88fe4a06 100644
--- a/summit/multiview_platform/utils/base.py
+++ b/summit/multiview_platform/utils/base.py
@@ -65,24 +65,24 @@ class BaseClassifier(BaseEstimator, ):
else:
return self.__class__.__name__ + " with no config."
- def get_base_estimator(self, base_estimator, estimator_config):
+ def get_base_estimator(self, estimator, estimator_config):
if estimator_config is None:
estimator_config = {}
if base_estimator is None:
return DecisionTreeClassifier(**estimator_config)
- if isinstance(base_estimator, str): # pragma: no cover
- if base_estimator == "DecisionTreeClassifier":
+ if isinstance(estimator, str): # pragma: no cover
+ if estimator == "DecisionTreeClassifier":
return DecisionTreeClassifier(**estimator_config)
- elif base_estimator == "AdaboostClassifier":
+ elif estimator == "AdaboostClassifier":
return AdaBoostClassifier(**estimator_config)
- elif base_estimator == "RandomForestClassifier":
+ elif estimator == "RandomForestClassifier":
return RandomForestClassifier(**estimator_config)
else:
raise ValueError(
'Base estimator string {} does not match an available classifier.'.format(
base_estimator))
- elif isinstance(base_estimator, BaseEstimator):
- return base_estimator.set_params(**estimator_config)
+ elif isinstance(estimator, BaseEstimator):
+ return estimator.set_params(**estimator_config)
else:
raise ValueError(
'base_estimator must be either a string or a BaseEstimator child class, it is {}'.format(
--
GitLab