From e1707216839678b5791a33e075b6c447483e25f8 Mon Sep 17 00:00:00 2001
From: Baptiste Bauvin <baptiste.bauvin@lis-lab.fr>
Date: Sun, 31 Mar 2019 13:50:06 -0400
Subject: [PATCH] mincq train times saved
---
.../MonoviewClassifiers/MinCQGraalpy.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/multiview_platform/MonoMultiViewClassifiers/MonoviewClassifiers/MinCQGraalpy.py b/multiview_platform/MonoMultiViewClassifiers/MonoviewClassifiers/MinCQGraalpy.py
index 8ef03c82..6ff8deb0 100644
--- a/multiview_platform/MonoMultiViewClassifiers/MonoviewClassifiers/MinCQGraalpy.py
+++ b/multiview_platform/MonoMultiViewClassifiers/MonoviewClassifiers/MinCQGraalpy.py
@@ -206,6 +206,8 @@ class RegularizedBinaryMinCqClassifier(MinCqClassifier):
"""
def fit(self, X, y):
+ import time
+ beg = time.time()
# We first fit and learn the weights.
super().fit(X, y)
@@ -221,7 +223,8 @@ class RegularizedBinaryMinCqClassifier(MinCqClassifier):
# logger.debug("Reversing decision of a binary voter")
self.weights[i] *= -1
self.estimators_[i].reverse_decision()
-
+ end=time.time()
+ self.train_time = end-beg
return self
def _solve(self, X, y):
@@ -329,6 +332,7 @@ class MinCQGraalpy(RegularizedBinaryMinCqClassifier, BaseMonoviewClassifier):
def getInterpret(self, directory, y_test):
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)
# y_rework = np.copy(y_test)
# y_rework[np.where(y_rework==0)] = -1
--
GitLab