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 /data | |
| parent | 4cfba3efb678bc83c9528d603244310fec66c0c0 (diff) | |
| download | taxi-0fd3b1497ffa1bb625bf593c845e28901bc640b7.tar.gz taxi-0fd3b1497ffa1bb625bf593c845e28901bc640b7.zip | |
Model with only embeddings.
Diffstat (limited to 'data')
| -rw-r--r-- | data/__init__.py | 1 | ||||
| -rw-r--r-- | data/transformers.py | 2 |
2 files changed, 2 insertions, 1 deletions
diff --git a/data/__init__.py b/data/__init__.py index 2121033..604809c 100644 --- a/data/__init__.py +++ b/data/__init__.py @@ -5,6 +5,7 @@ import numpy path = os.environ.get('TAXI_PATH', '/data/lisatmp3/auvolat/taxikaggle') +path = os.environ.get('TAXI_PATH', '/Users/adeb/data/taxi') Polyline = h5py.special_dtype(vlen=numpy.float32) diff --git a/data/transformers.py b/data/transformers.py index 1bed887..e6806cc 100644 --- a/data/transformers.py +++ b/data/transformers.py @@ -94,7 +94,7 @@ class TaxiGenerateSplits(Transformer): class _taxi_add_first_last_len_helper(object): - def __init__(self, k, latitude, longitude): + def __init__(self, k, id_latitude, id_longitude): self.k = k self.id_latitude = id_latitude self.id_longitude = id_longitude |