From 1ff071800fc876eb6f2c25fe0eb1f7dc64efe0be Mon Sep 17 00:00:00 2001 From: Alex Auvolat Date: Wed, 6 May 2015 11:56:01 -0400 Subject: Add Bokeh plotting --- train.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'train.py') diff --git a/train.py b/train.py index 54b3da3..40449f0 100755 --- 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), -- cgit v1.2.3