diff options
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 |