aboutsummaryrefslogtreecommitdiff
path: root/model.py
diff options
context:
space:
mode:
authorAlex Auvolat <alex.auvolat@ens.fr>2015-05-04 16:43:48 -0400
committerAlex Auvolat <alex.auvolat@ens.fr>2015-05-04 16:58:31 -0400
commit80d3ea67a845484d119cb88f0a0412f981ab344c (patch)
tree37b8130b6d761bcda48c8c0f74114498b85dad97 /model.py
parentf9a31bd246e3c4736d3f532b566b7437eba6b4de (diff)
downloadtaxi-80d3ea67a845484d119cb88f0a0412f981ab344c.tar.gz
taxi-80d3ea67a845484d119cb88f0a0412f981ab344c.zip
Mew data analysis tool: clustering of arrival points.
Diffstat (limited to 'model.py')
-rw-r--r--model.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/model.py b/model.py
index 753fb01..744d877 100644
--- a/model.py
+++ b/model.py
@@ -53,7 +53,7 @@ def setup_train_stream():
load_in_memory=True)
train = DataStream(train, iteration_scheme=SequentialExampleScheme(data.dataset_size - config.n_valid))
train = transformers.filter_out_trips(data.valid_trips, train)
- train = transformers.TaxiGenerateSplits(train)
+ train = transformers.TaxiGenerateSplits(train, max_splits=100)
train = transformers.add_first_k(config.n_begin_end_pts, train)
train = transformers.add_last_k(config.n_begin_end_pts, train)
train = transformers.Select(train, ('origin_stand', 'origin_call', 'first_k_latitude',