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