taxi

commit 1ff071800fc876eb6f2c25fe0eb1f7dc64efe0be
parent 0b4b65cb3d88ac4818e71ccef0bded3ddee0683c
Author: Alex Auvolat <alex.auvolat@ens.fr>
Date:   Wed,  6 May 2015 11:56:01 -0400

Add Bokeh plotting

Diffstat:
M
train.py
 | 
2
++

1 file changed, 2 insertions(+), 0 deletions(-)
diff --git a/train.py b/train.py
@@ -18,6 +18,7 @@ from blocks.main_loop import MainLoop
 from blocks.extensions import Printing, FinishAfter
 from blocks.extensions.saveload import Dump, LoadFromDump, Checkpoint
 from blocks.extensions.monitoring import DataStreamMonitoring, TrainingDataMonitoring
+from blocks.extensions.plot import Plot
 
 from data import transformers
 from data.hdf5 import TaxiDataset, TaxiStream
@@ -103,6 +104,7 @@ def main():
                                      prefix='valid',
                                      every_n_batches=1000),
                 Printing(every_n_batches=1000),
+                Plot(model_name, channels=[['valid_cost']], every_n_batches=1000),
                 # Checkpoint('model.pkl', every_n_batches=100),
                 Dump('model_data/' + model_name, every_n_batches=1000),
                 LoadFromDump('model_data/' + model_name),