aboutsummaryrefslogtreecommitdiff
path: root/data/transformers.py
diff options
context:
space:
mode:
authorAlex Auvolat <alex.auvolat@ens.fr>2015-05-06 10:12:17 -0400
committerAlex Auvolat <alex.auvolat@ens.fr>2015-05-06 10:12:17 -0400
commit35b4503ddd148b0c937468891dd0a7e9ff1c79f4 (patch)
tree92db398bce1f557ca9cb988dcf64a700ad249e47 /data/transformers.py
parent60aa0d3fdd42d7489cc69acbb54c59d7c249ea34 (diff)
downloadtaxi-35b4503ddd148b0c937468891dd0a7e9ff1c79f4.tar.gz
taxi-35b4503ddd148b0c937468891dd0a7e9ff1c79f4.zip
Move weights init to config files ; fix s/time/travel_time
Diffstat (limited to 'data/transformers.py')
-rw-r--r--data/transformers.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/data/transformers.py b/data/transformers.py
index 1cc4834..e3ff7b1 100644
--- a/data/transformers.py
+++ b/data/transformers.py
@@ -34,7 +34,7 @@ class Select(Transformer):
class TaxiGenerateSplits(Transformer):
def __init__(self, data_stream, max_splits=-1):
super(TaxiGenerateSplits, self).__init__(data_stream)
- self.sources = data_stream.sources + ('destination_latitude', 'destination_longitude', 'time')
+ self.sources = data_stream.sources + ('destination_latitude', 'destination_longitude', 'travel_time')
self.max_splits = max_splits
self.data = None
self.splits = []