diff options
Diffstat (limited to 'model/mlp.py')
| -rw-r--r-- | model/mlp.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/model/mlp.py b/model/mlp.py index 6abc86f..a6f3991 100644 --- a/model/mlp.py +++ b/model/mlp.py @@ -63,6 +63,7 @@ class Stream(object): stream = transformers.TaxiExcludeTrips(stream, valid_trips_ids) stream = transformers.TaxiGenerateSplits(stream, max_splits=self.config.max_splits) + stream = transformers.add_destination(stream) stream = transformers.taxi_add_datetime(stream) stream = transformers.taxi_add_first_last_len(stream, self.config.n_begin_end_pts) |