From bd08e452093bba68fe2d79b1e9da76488b203720 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=89tienne=20Simon?= Date: Mon, 22 Jun 2015 14:40:19 -0400 Subject: Update memory network --- model/memory_network.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'model') diff --git a/model/memory_network.py b/model/memory_network.py index 92e83e2..5acbfe3 100644 --- a/model/memory_network.py +++ b/model/memory_network.py @@ -88,7 +88,7 @@ class Stream(object): dataset = TaxiDataset('train') - prefix_stream = DataStream(dataset, iteration_scheme=TaxiTimeCutScheme()) + prefix_stream = DataStream(dataset, iteration_scheme=TaxiTimeCutScheme(self.config.num_cuts)) prefix_stream = transformers.TaxiExcludeTrips(prefix_stream, valid_trips_ids) prefix_stream = transformers.TaxiGenerateSplits(prefix_stream, max_splits=self.config.max_splits) prefix_stream = transformers.taxi_add_datetime(prefix_stream) -- cgit v1.2.3