From f6a627ca87612dd8a1650d78910a8a37600e9545 Mon Sep 17 00:00:00 2001
From: Baptiste Bauvin <baptiste.bauvin@lis-lab.fr>
Date: Fri, 15 Feb 2019 08:22:12 -0500
Subject: [PATCH] Added result saving for cqboost
---
.../MonoviewClassifiers/CQBoost.py | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/multiview_platform/MonoMultiViewClassifiers/MonoviewClassifiers/CQBoost.py b/multiview_platform/MonoMultiViewClassifiers/MonoviewClassifiers/CQBoost.py
index ea3c4f1e..68497d53 100644
--- a/multiview_platform/MonoMultiViewClassifiers/MonoviewClassifiers/CQBoost.py
+++ b/multiview_platform/MonoMultiViewClassifiers/MonoviewClassifiers/CQBoost.py
@@ -24,6 +24,16 @@ class CQBoost(ColumnGenerationClassifier, BaseMonoviewClassifier):
def getInterpret(self, directory, y_test):
np.savetxt(directory + "train_metrics.csv", self.train_metrics, delimiter=',')
+ np.savetxt(directory + "y_test_step.csv", self.step_decisions,
+ delimiter=',')
+ step_metrics = []
+ for step_index in range(self.step_decisions.shape[1] - 1):
+ step_metrics.append(self.plotted_metric.score(y_test,
+ self.step_decisions[:,
+ step_index]))
+ step_metrics = np.array(step_metrics)
+ np.savetxt(directory + "step_test_metrics.csv", step_metrics,
+ delimiter=',')
return getInterpretBase(self, directory, "CQBoost", self.weights_, y_test)
--
GitLab