From 5750054343a7cdfd39d7f01cf01068e9183f9384 Mon Sep 17 00:00:00 2001 From: Baptiste Bauvin <baptiste.bauvin@lis-lab.fr> Date: Wed, 20 May 2020 09:04:39 -0400 Subject: [PATCH] None config possible --- summit/multiview_platform/monoview_classifiers/adaboost.py | 1 + summit/multiview_platform/utils/base.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/summit/multiview_platform/monoview_classifiers/adaboost.py b/summit/multiview_platform/monoview_classifiers/adaboost.py index 82b380f7..cd8ce3db 100644 --- a/summit/multiview_platform/monoview_classifiers/adaboost.py +++ b/summit/multiview_platform/monoview_classifiers/adaboost.py @@ -40,6 +40,7 @@ class Adaboost(AdaBoostClassifier, BaseMonoviewClassifier): self.weird_strings = {"base_estimator": "class_name"} self.plotted_metric = metrics.zero_one_loss self.plotted_metric_name = "zero_one_loss" + self.base_estimator_config = base_estimator_config self.step_predictions = None def fit(self, X, y, sample_weight=None): diff --git a/summit/multiview_platform/utils/base.py b/summit/multiview_platform/utils/base.py index 8dcaaf81..529ae1e9 100644 --- a/summit/multiview_platform/utils/base.py +++ b/summit/multiview_platform/utils/base.py @@ -99,6 +99,8 @@ class BaseClassifier(BaseEstimator, ): if "config" in self.weird_strings[param_name]: string += "( with " + self.get_params()[ param_name].params_to_string() + ")" + elif self.get_params()[param_name] is None: + return "None" else: return str(self.get_params()[param_name]) -- GitLab