diff options
| author | Alex Auvolat <alex.auvolat@ens.fr> | 2015-06-11 15:26:21 -0400 |
|---|---|---|
| committer | Alex Auvolat <alex.auvolat@ens.fr> | 2015-06-11 15:26:21 -0400 |
| commit | c3ec750f4a04e3117d658e8275dd3d91d2b0cbe4 (patch) | |
| tree | 6fb3fcc9425948a13fdc421534b7a00574c1e595 /model/mlp.py | |
| parent | 7e8dbac77ce712846954bdd5f4bfb62b6efaf7df (diff) | |
| download | taxi-c3ec750f4a04e3117d658e8275dd3d91d2b0cbe4.tar.gz taxi-c3ec750f4a04e3117d658e8275dd3d91d2b0cbe4.zip | |
Add TaxiRemoveTestOnlyClients ; custom dumpmanager enabling multiprocessing
Diffstat (limited to 'model/mlp.py')
| -rw-r--r-- | model/mlp.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/model/mlp.py b/model/mlp.py index 2e0b9e5..b1e9163 100644 --- a/model/mlp.py +++ b/model/mlp.py @@ -70,6 +70,8 @@ class Stream(object): stream = Batch(stream, iteration_scheme=ConstantScheme(self.config.batch_size)) + stream = MultiProcessing(stream) + return stream def valid(self, req_vars): @@ -85,6 +87,7 @@ class Stream(object): stream = transformers.TaxiAddDateTime(stream) stream = transformers.TaxiAddFirstLastLen(self.config.n_begin_end_pts, stream) + stream = transformers.TaxiRemoveTestOnlyClients(stream) return Batch(stream, iteration_scheme=ConstantScheme(1)) |