From 757c017056120d1eb484f137d60c68eb660ebd94 Mon Sep 17 00:00:00 2001
From: Baptiste Bauvin <baptiste.bauvin@lis-lab.fr>
Date: Wed, 8 Aug 2018 12:21:29 -0400
Subject: [PATCH] Adaboost ckecked with sklearn

---
 .../MonoviewClassifiers/Adaboost.py                       | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/multiview_platform/MonoMultiViewClassifiers/MonoviewClassifiers/Adaboost.py b/multiview_platform/MonoMultiViewClassifiers/MonoviewClassifiers/Adaboost.py
index 9980f721..05f6545b 100644
--- a/multiview_platform/MonoMultiViewClassifiers/MonoviewClassifiers/Adaboost.py
+++ b/multiview_platform/MonoMultiViewClassifiers/MonoviewClassifiers/Adaboost.py
@@ -10,8 +10,8 @@ __status__ = "Prototype"  # Production, Development, Prototype
 
 class Adaboost(AdaBoostClassifier, BaseMonoviewClassifier):
 
-    def __init__(self, random_state, n_estimators=10,
-                 base_estimator=DecisionTreeClassifier(), **kwargs):
+    def __init__(self, random_state=None, n_estimators=10,
+                 base_estimator=None, **kwargs):
         super(Adaboost, self).__init__(
             random_state=random_state,
             n_estimators=n_estimators,
@@ -19,7 +19,7 @@ class Adaboost(AdaBoostClassifier, BaseMonoviewClassifier):
             )
         self.param_names = ["n_estimators", "base_estimator"]
         self.classed_params = ["base_estimator"]
-        self.distribs = [CustomRandint(low=1, high=500), [DecisionTreeClassifier()]]
+        self.distribs = [CustomRandint(low=1, high=500), [None]]
         self.weird_strings = {"base_estimator":"class_name"}
 
     def canProbas(self):
@@ -31,7 +31,7 @@ class Adaboost(AdaBoostClassifier, BaseMonoviewClassifier):
         paramsSet = []
         for _ in range(nIter):
             paramsSet.append({"n_estimators": self.random_state.randint(1, 150),
-                              "base_estimator": DecisionTreeClassifier()})
+                              "base_estimator": None})
         return paramsSet
 
     def getInterpret(self, directory):
-- 
GitLab