diff options
| author | Alex Auvolat <alex@adnab.me> | 2015-07-10 19:21:04 -0400 |
|---|---|---|
| committer | Alex Auvolat <alex@adnab.me> | 2015-07-10 19:21:04 -0400 |
| commit | 8442ba13a442d509a407b5913b95d9f1b00989d2 (patch) | |
| tree | 821568570dcc277c1638c72ad06460b2a7b03b05 /config/dest_mlp_tgtcls_1_cswdtx_batchshuffle.py | |
| parent | c5187418bc93c34e3fdce4fdc1a3b5316812b69a (diff) | |
| parent | 97b9672860b97a397ae48b24287741922e7fcb8a (diff) | |
| download | taxi-8442ba13a442d509a407b5913b95d9f1b00989d2.tar.gz taxi-8442ba13a442d509a407b5913b95d9f1b00989d2.zip | |
Merge branch 'master' of github.com:adbrebs/taxi
Diffstat (limited to 'config/dest_mlp_tgtcls_1_cswdtx_batchshuffle.py')
| -rw-r--r-- | config/dest_mlp_tgtcls_1_cswdtx_batchshuffle.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/config/dest_mlp_tgtcls_1_cswdtx_batchshuffle.py b/config/dest_mlp_tgtcls_1_cswdtx_batchshuffle.py index a4db33c..b816930 100644 --- a/config/dest_mlp_tgtcls_1_cswdtx_batchshuffle.py +++ b/config/dest_mlp_tgtcls_1_cswdtx_batchshuffle.py @@ -23,14 +23,14 @@ dim_embeddings = [ ] dim_input = n_begin_end_pts * 2 * 2 + sum(x for (_, _, x) in dim_embeddings) -dim_hidden = [1000] +dim_hidden = [500] dim_output = tgtcls.shape[0] embed_weights_init = IsotropicGaussian(0.01) mlp_weights_init = IsotropicGaussian(0.1) mlp_biases_init = Constant(0.01) -step_rule = Momentum(learning_rate=0.01, momentum=0.9) +step_rule = Momentum(learning_rate=0.001, momentum=0.99) batch_size = 200 |