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): ...@@ -331,8 +331,10 @@ class MinCQGraalpy(RegularizedBinaryMinCqClassifier, BaseMonoviewClassifier):
self.param_names = ["mu"] self.param_names = ["mu"]
self.distribs = [CustomUniform(loc=0.5, state=2.0, multiplier="e-"), self.distribs = [CustomUniform(loc=0.5, state=2.0, multiplier="e-"),
] ]
self.n_stumps_per_attribute = n_stumps_per_attribute
self.classed_params = [] self.classed_params = []
self.weird_strings = {} self.weird_strings = {}
self.random_state = random_state
if "nbCores" not in kwargs: if "nbCores" not in kwargs:
self.nbCores = 1 self.nbCores = 1
else: else:
...@@ -342,6 +344,12 @@ class MinCQGraalpy(RegularizedBinaryMinCqClassifier, BaseMonoviewClassifier): ...@@ -342,6 +344,12 @@ class MinCQGraalpy(RegularizedBinaryMinCqClassifier, BaseMonoviewClassifier):
"""Used to know if the classifier can return label probabilities""" """Used to know if the classifier can return label probabilities"""
return True 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): def getInterpret(self, directory, y_test):
interpret_string = "" interpret_string = ""
# interpret_string += "Train C_bound value : "+str(self.cbound_train) # interpret_string += "Train C_bound value : "+str(self.cbound_train)
......
...@@ -125,7 +125,7 @@ def parseTheArgs(arguments): ...@@ -125,7 +125,7 @@ def parseTheArgs(arguments):
groupAdaboostPregen = parser.add_argument_group('AdaboostPregen arguments') groupAdaboostPregen = parser.add_argument_group('AdaboostPregen arguments')
groupAdaboostPregen.add_argument('--AdP_n_est', metavar='INT', type=int, groupAdaboostPregen.add_argument('--AdP_n_est', metavar='INT', type=int,
action='store', help='Number of estimators', action='store', help='Number of estimators',
default=2) default=100)
groupAdaboostPregen.add_argument('--AdP_b_est', metavar='STRING', action='store', groupAdaboostPregen.add_argument('--AdP_b_est', metavar='STRING', action='store',
help='Estimators', help='Estimators',
default='DecisionTreeClassifier') default='DecisionTreeClassifier')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment