From 14b44192663e43e8afda83d4d936738fe9592db0 Mon Sep 17 00:00:00 2001
From: Baptiste Bauvin <baptiste.bauvin@lis-lab.fr>
Date: Tue, 19 Mar 2019 08:37:30 -0400
Subject: [PATCH] Added std in csv
---
.../MonoMultiViewClassifiers/ResultAnalysis.py | 12 +++++++++++-
1 file changed, 11 insertions(+), 1 deletion(-)
diff --git a/multiview_platform/MonoMultiViewClassifiers/ResultAnalysis.py b/multiview_platform/MonoMultiViewClassifiers/ResultAnalysis.py
index 9ef17e0d..4f97eb04 100644
--- a/multiview_platform/MonoMultiViewClassifiers/ResultAnalysis.py
+++ b/multiview_platform/MonoMultiViewClassifiers/ResultAnalysis.py
@@ -244,7 +244,17 @@ def plotMetricScores(trainScores, testScores, names, nbResults, metricName, file
f.savefig(fileName+'.png')
plt.close()
import pandas as pd
- dataframe = pd.DataFrame(np.transpose(np.concatenate((trainScores.reshape((trainScores.shape[0], 1)), testScores.reshape((trainScores.shape[0], 1))), axis=1)), columns=names)
+ if train_STDs is None :
+ dataframe = pd.DataFrame(np.transpose(np.concatenate((
+ trainScores.reshape((trainScores.shape[0], 1)),
+ testScores.reshape((trainScores.shape[0], 1))), axis=1)),
+ columns=names)
+ else:
+ dataframe = pd.DataFrame(np.transpose(np.concatenate((
+ trainScores.reshape((trainScores.shape[0], 1)),
+ train_STDs.reshape((trainScores.shape[0], 1)),
+ testScores.reshape((trainScores.shape[0], 1)),
+ test_STDs.reshape((trainScores.shape[0], 1))), axis=1)), columns=names)
dataframe.to_csv(fileName+".csv")
--
GitLab