From 80d3ea67a845484d119cb88f0a0412f981ab344c Mon Sep 17 00:00:00 2001 From: Alex Auvolat Date: Mon, 4 May 2015 16:43:48 -0400 Subject: Mew data analysis tool: clustering of arrival points. --- model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'model.py') 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', -- cgit v1.2.3