From 432705f87aa69ceb986550e8437badbf4f5871c1 Mon Sep 17 00:00:00 2001 From: Baptiste Bauvin <baptiste.bauvin@lis-lab.fr> Date: Mon, 2 Mar 2020 15:28:19 +0100 Subject: [PATCH] Allowed one classifier for all views in late fusion --- .../multiview_classifiers/additions/late_fusion_utils.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/multiview_platform/mono_multi_view_classifiers/multiview_classifiers/additions/late_fusion_utils.py b/multiview_platform/mono_multi_view_classifiers/multiview_classifiers/additions/late_fusion_utils.py index d6ff8b4c..f30aa6a4 100644 --- a/multiview_platform/mono_multi_view_classifiers/multiview_classifiers/additions/late_fusion_utils.py +++ b/multiview_platform/mono_multi_view_classifiers/multiview_classifiers/additions/late_fusion_utils.py @@ -137,11 +137,15 @@ class LateFusionClassifier(BaseMultiviewClassifier, BaseFusionClassifier): nb_clfs = nb_monoview_per_view else: nb_clfs = nb_view + if isinstance(self.classifiers_names, ClassifierDistribution): self.classifiers_names = self.classifiers_names.draw(nb_clfs, self.rs) elif self.classifiers_names is None: self.classifiers_names = ["decision_tree" for _ in range(nb_clfs)] + elif isinstance(self.classifiers_names, str): + self.classifiers_names = [self.classifiers_names + for _ in range(nb_clfs)] if isinstance(self.classifier_configs, ConfigDistribution): self.classifier_configs = self.classifier_configs.draw(nb_clfs, -- GitLab