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