From f9bd449b06815cfba0e250bd648c4e740f8b5857 Mon Sep 17 00:00:00 2001
From: Baptiste Bauvin <baptiste.bauvin@lis-lab.fr>
Date: Wed, 27 Mar 2019 12:01:56 -0400
Subject: [PATCH] mioncqgp randomsearch compatible

---
 .../MonoviewClassifiers/MinCQGraalpy.py                   | 8 ++++++++
 .../MonoMultiViewClassifiers/utils/execution.py           | 2 +-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/multiview_platform/MonoMultiViewClassifiers/MonoviewClassifiers/MinCQGraalpy.py b/multiview_platform/MonoMultiViewClassifiers/MonoviewClassifiers/MinCQGraalpy.py
index b638c6fb..b1bf1dfe 100644
--- a/multiview_platform/MonoMultiViewClassifiers/MonoviewClassifiers/MinCQGraalpy.py
+++ b/multiview_platform/MonoMultiViewClassifiers/MonoviewClassifiers/MinCQGraalpy.py
@@ -331,8 +331,10 @@ class MinCQGraalpy(RegularizedBinaryMinCqClassifier, BaseMonoviewClassifier):
         self.param_names = ["mu"]
         self.distribs = [CustomUniform(loc=0.5, state=2.0, multiplier="e-"),
                          ]
+        self.n_stumps_per_attribute = n_stumps_per_attribute
         self.classed_params = []
         self.weird_strings = {}
+        self.random_state = random_state
         if "nbCores" not in kwargs:
             self.nbCores = 1
         else:
@@ -342,6 +344,12 @@ class MinCQGraalpy(RegularizedBinaryMinCqClassifier, BaseMonoviewClassifier):
         """Used to know if the classifier can return label probabilities"""
         return True
 
+    def set_params(self, **params):
+        self.mu = params["mu"]
+
+    def get_params(self, deep=True):
+        return {"random_state":self.random_state, "mu":self.mu}
+
     def getInterpret(self, directory, y_test):
         interpret_string = ""
         # interpret_string += "Train C_bound value : "+str(self.cbound_train)
diff --git a/multiview_platform/MonoMultiViewClassifiers/utils/execution.py b/multiview_platform/MonoMultiViewClassifiers/utils/execution.py
index 9a3a91d4..2fc233f2 100644
--- a/multiview_platform/MonoMultiViewClassifiers/utils/execution.py
+++ b/multiview_platform/MonoMultiViewClassifiers/utils/execution.py
@@ -125,7 +125,7 @@ def parseTheArgs(arguments):
     groupAdaboostPregen = parser.add_argument_group('AdaboostPregen arguments')
     groupAdaboostPregen.add_argument('--AdP_n_est', metavar='INT', type=int,
                                action='store', help='Number of estimators',
-                               default=2)
+                               default=100)
     groupAdaboostPregen.add_argument('--AdP_b_est', metavar='STRING', action='store',
                                help='Estimators',
                                default='DecisionTreeClassifier')
-- 
GitLab