diff options
| author | AdeB <adbrebs@gmail.com> | 2015-06-21 17:00:25 -0400 |
|---|---|---|
| committer | AdeB <adbrebs@gmail.com> | 2015-06-21 17:00:25 -0400 |
| commit | 0fd3b1497ffa1bb625bf593c845e28901bc640b7 (patch) | |
| tree | f03e58cbf9603d0bb3c32060e54d9756aa018048 /model/mlp.py | |
| parent | 4cfba3efb678bc83c9528d603244310fec66c0c0 (diff) | |
| download | taxi-0fd3b1497ffa1bb625bf593c845e28901bc640b7.tar.gz taxi-0fd3b1497ffa1bb625bf593c845e28901bc640b7.zip | |
Model with only embeddings.
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) |