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

mincq train times saved

parent 59bf3b4b
Branches
No related tags found
No related merge requests found
...@@ -206,6 +206,8 @@ class RegularizedBinaryMinCqClassifier(MinCqClassifier): ...@@ -206,6 +206,8 @@ class RegularizedBinaryMinCqClassifier(MinCqClassifier):
""" """
def fit(self, X, y): def fit(self, X, y):
import time
beg = time.time()
# We first fit and learn the weights. # We first fit and learn the weights.
super().fit(X, y) super().fit(X, y)
...@@ -221,7 +223,8 @@ class RegularizedBinaryMinCqClassifier(MinCqClassifier): ...@@ -221,7 +223,8 @@ class RegularizedBinaryMinCqClassifier(MinCqClassifier):
# logger.debug("Reversing decision of a binary voter") # logger.debug("Reversing decision of a binary voter")
self.weights[i] *= -1 self.weights[i] *= -1
self.estimators_[i].reverse_decision() self.estimators_[i].reverse_decision()
end=time.time()
self.train_time = end-beg
return self return self
def _solve(self, X, y): def _solve(self, X, y):
...@@ -329,6 +332,7 @@ class MinCQGraalpy(RegularizedBinaryMinCqClassifier, BaseMonoviewClassifier): ...@@ -329,6 +332,7 @@ class MinCQGraalpy(RegularizedBinaryMinCqClassifier, BaseMonoviewClassifier):
def getInterpret(self, directory, y_test): def getInterpret(self, directory, y_test):
interpret_string = "Cbound on train :"+str(self.train_cbound) interpret_string = "Cbound on train :"+str(self.train_cbound)
np.savetxt(directory+"times.csv", np.array(self.train_time))
# interpret_string += "Train C_bound value : "+str(self.cbound_train) # interpret_string += "Train C_bound value : "+str(self.cbound_train)
# y_rework = np.copy(y_test) # y_rework = np.copy(y_test)
# y_rework[np.where(y_rework==0)] = -1 # y_rework[np.where(y_rework==0)] = -1
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment