diff options
| author | Alex Auvolat <alex.auvolat@ens.fr> | 2015-07-23 19:00:52 -0400 |
|---|---|---|
| committer | Alex Auvolat <alex.auvolat@ens.fr> | 2015-07-23 19:01:13 -0400 |
| commit | 1795dfe742bcb75085a909413b723b64a8eeb4fc (patch) | |
| tree | c1689d428e8dcbf0e81411175290b0bcaee9938c /model/bidirectional.py | |
| parent | fe608831c62c7dba60a3bf57433d97b999e567c8 (diff) | |
| download | taxi-1795dfe742bcb75085a909413b723b64a8eeb4fc.tar.gz taxi-1795dfe742bcb75085a909413b723b64a8eeb4fc.zip | |
Memory network with bidirectionnal RNN
Diffstat (limited to 'model/bidirectional.py')
| -rw-r--r-- | model/bidirectional.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/model/bidirectional.py b/model/bidirectional.py index af3891d..483ea54 100644 --- a/model/bidirectional.py +++ b/model/bidirectional.py @@ -121,8 +121,8 @@ class Stream(object): stream = transformers.balanced_batch(stream, key='latitude', batch_size=self.config.batch_size, batch_sort_size=self.config.batch_sort_size) stream = Padding(stream, mask_sources=['latitude', 'longitude']) + stream = transformers.Select(stream, req_vars) - stream = MultiProcessing(stream) return stream |