From 173d5762bc461f7396b5e256ceb37c8053da194f Mon Sep 17 00:00:00 2001 From: reedts Date: Mon, 13 Aug 2018 18:54:34 +0200 Subject: [PATCH] Fix multiple graphs and plot while evaluating --- pywatts/eval_training.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/pywatts/eval_training.py b/pywatts/eval_training.py index 22ddd42..ab95090 100644 --- a/pywatts/eval_training.py +++ b/pywatts/eval_training.py @@ -54,15 +54,21 @@ if TRAIN: pp.plot(pred_result, 'black') pp.plot(prediction, color=color_gradient_base) + pp.savefig(FIGURE_OUTPUT_DIR+'{}.pdf'.format(q), orientation='landscape') color_gradient_base = tuple([sum(x) for x in zip(color_gradient_base, color_step_width)]) for i in range(NUM_QUERIES): - pp.figure(i) - pp.savefig(FIGURE_OUTPUT_DIR+'{}.pdf'.format(i), orientation='landscape') + pp.close(i) if PLOT: # Plot training success rate (with 'average loss') - pywatts.routines.plot_training(train_eval) + loss = [] + for e in train_eval: + loss.append(e['average_loss']) + + pp.plot(loss) + # Needed for execution in PyCharm + pp.show() exit()