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