taxi

commit 389bfd3637dfb523a3e4194c7281a0c538166546
parent 13fc171f60ae1981c7ad4f2a302a8a85c29addc5
Author: Étienne Simon <esimon@esimon.eu>
Date:   Thu, 23 Jul 2015 22:15:08 -0400

Fix RNN Stream without tvt

Diffstat:
M
model/rnn.py
 | 
2
++

1 file changed, 2 insertions(+), 0 deletions(-)
diff --git a/model/rnn.py b/model/rnn.py
@@ -155,6 +155,8 @@ class Stream(object):
 
         stream = transformers.TaxiExcludeEmptyTrips(stream)
         stream = transformers.taxi_add_datetime(stream)
+        if not data.tvt:
+            stream = transformers.add_destination(stream)
         stream = transformers.Select(stream, tuple(v for v in req_vars if not v.endswith('_mask')))
 
         stream = transformers.balanced_batch(stream, key='latitude', batch_size=self.config.batch_size, batch_sort_size=self.config.batch_sort_size)