From 0fd3b1497ffa1bb625bf593c845e28901bc640b7 Mon Sep 17 00:00:00 2001 From: AdeB Date: Sun, 21 Jun 2015 17:00:25 -0400 Subject: Model with only embeddings. --- data/__init__.py | 1 + data/transformers.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) (limited to 'data') 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 -- cgit v1.2.3