diff options
| author | Étienne Simon <esimon@esimon.eu> | 2015-06-22 14:40:19 -0400 |
|---|---|---|
| committer | Étienne Simon <esimon@esimon.eu> | 2015-06-22 14:40:19 -0400 |
| commit | bd08e452093bba68fe2d79b1e9da76488b203720 (patch) | |
| tree | 48d82d8cf883ad642c483e2cad5cc707dd8c4694 /model/memory_network.py | |
| parent | ad5a03c6f60e5b2d543326bf8917b48e5b390b82 (diff) | |
| download | taxi-bd08e452093bba68fe2d79b1e9da76488b203720.tar.gz taxi-bd08e452093bba68fe2d79b1e9da76488b203720.zip | |
Update memory network
Diffstat (limited to 'model/memory_network.py')
| -rw-r--r-- | model/memory_network.py | 2 |
1 files changed, 1 insertions, 1 deletions
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) |