Skip to content
Snippets Groups Projects
Commit f9bd449b authored by Baptiste Bauvin's avatar Baptiste Bauvin
Browse files

mioncqgp randomsearch compatible

parent f875ec80
No related branches found
No related tags found
No related merge requests found
......@@ -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)
......
......@@ -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')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment