vanitasvitae/pywatts
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Merge branch 'master' of github.com:vanitasvitae/pywatts

Browse source
This commit is contained in:
reedts 2018-08-14 15:22:07 +02:00
commit dfddb8799e
3 changed files with 13 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
.gitignore vendored
View file

@ -114,3 +114,6 @@ tf_pywatts_model/
# Tensorboard # Tensorboard
pywatts/tensorboard pywatts/tensorboard
# Figures
figures/

12
pywatts/eval_training.py
View file

@ -54,15 +54,21 @@ if TRAIN:
pp.plot(pred_result, 'black') pp.plot(pred_result, 'black')
pp.plot(prediction, color=color_gradient_base) 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)]) color_gradient_base = tuple([sum(x) for x in zip(color_gradient_base, color_step_width)])
for i in range(NUM_QUERIES): for i in range(NUM_QUERIES):
pp.figure(i) pp.close(i)
pp.savefig(FIGURE_OUTPUT_DIR+'{}.pdf'.format(i), orientation='landscape')
if PLOT: if PLOT:
# Plot training success rate (with 'average loss') # 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() exit()

2
pywatts/test_kcross_train.py
View file

@ -35,7 +35,7 @@ if TRAIN:
if PLOT: if PLOT:
# Plot training success rate (with 'average loss') # Plot training success rate (with 'average loss')
pywatts.main.plot_training(train_eval) pywatts.routines.plot_training(train_eval)
exit() exit()