diff options
Diffstat (limited to 'model')
| -rw-r--r-- | model/bidirectional.py | 2 | ||||
| -rw-r--r-- | model/rnn.py | 1 |
2 files changed, 0 insertions, 3 deletions
diff --git a/model/bidirectional.py b/model/bidirectional.py index 1d697d6..149d1fd 100644 --- a/model/bidirectional.py +++ b/model/bidirectional.py @@ -116,7 +116,6 @@ class Stream(object): stream = Unpack(stream) stream = transformers.taxi_add_datetime(stream) - 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) @@ -131,7 +130,6 @@ class Stream(object): stream = TaxiStream(self.config.valid_set, 'valid.hdf5') stream = transformers.taxi_add_datetime(stream) - stream = transformers.add_destination(stream) stream = transformers.Select(stream, tuple(v for v in req_vars if not v.endswith('_mask'))) stream = Batch(stream, iteration_scheme=ConstantScheme(self.config.batch_size)) diff --git a/model/rnn.py b/model/rnn.py index d3f6616..70d054a 100644 --- a/model/rnn.py +++ b/model/rnn.py @@ -165,7 +165,6 @@ class Stream(object): def valid(self, req_vars): stream = TaxiStream(self.config.valid_set, 'valid.hdf5') stream = transformers.taxi_add_datetime(stream) - stream = transformers.add_destination(stream) stream = transformers.Select(stream, tuple(v for v in req_vars if not v.endswith('_mask'))) stream = Batch(stream, iteration_scheme=ConstantScheme(self.config.batch_size)) |