From 5b02ecbc0012db8ec72372324b014bd06622b209 Mon Sep 17 00:00:00 2001
From: Baptiste Bauvin <baptiste.bauvin@lis-lab.fr>
Date: Fri, 20 Mar 2020 09:25:37 -0400
Subject: [PATCH] adaboost same size train test bug setteleed
---
.../monoview_classifiers/adaboost.py | 6 ++----
1 file changed, 2 insertions(+), 4 deletions(-)
diff --git a/multiview_platform/mono_multi_view_classifiers/monoview_classifiers/adaboost.py b/multiview_platform/mono_multi_view_classifiers/monoview_classifiers/adaboost.py
index 8f5e55a3..6fbfbd8b 100644
--- a/multiview_platform/mono_multi_view_classifiers/monoview_classifiers/adaboost.py
+++ b/multiview_platform/mono_multi_view_classifiers/monoview_classifiers/adaboost.py
@@ -124,10 +124,8 @@ class Adaboost(AdaBoostClassifier, BaseMonoviewClassifier):
pred = AdaBoostClassifier.predict(self, X)
end = time.time()
self.pred_time = end - begin
- # TODO : mauvaise verif
- if X.shape != self.train_shape:
- self.step_predictions = np.array(
- [step_pred for step_pred in self.staged_predict(X)])
+ self.step_predictions = np.array(
+ [step_pred for step_pred in self.staged_predict(X)])
return pred
def get_interpretation(self, directory, base_file_name, y_test, multi_class=False):
--
GitLab