diff options
| author | Alex Auvolat <alex.auvolat@ens.fr> | 2015-07-02 15:00:51 -0400 |
|---|---|---|
| committer | Alex Auvolat <alex.auvolat@ens.fr> | 2015-07-02 15:00:51 -0400 |
| commit | 66d2717188e189fde5422576740903ca8e488f63 (patch) | |
| tree | 555f94f67fe061b1d6fb9977b4c9559d6a16898a /config/dest_mlp_tgtcls_1_cswdtx_alexandre.py | |
| parent | 6cf80aff3e4dc57d13a33c2946bc0ae57cfae6b8 (diff) | |
| download | taxi-66d2717188e189fde5422576740903ca8e488f63.tar.gz taxi-66d2717188e189fde5422576740903ca8e488f63.zip | |
Add small models
Diffstat (limited to 'config/dest_mlp_tgtcls_1_cswdtx_alexandre.py')
| -rw-r--r-- | config/dest_mlp_tgtcls_1_cswdtx_alexandre.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/config/dest_mlp_tgtcls_1_cswdtx_alexandre.py b/config/dest_mlp_tgtcls_1_cswdtx_alexandre.py index d75810a..db69cd3 100644 --- a/config/dest_mlp_tgtcls_1_cswdtx_alexandre.py +++ b/config/dest_mlp_tgtcls_1_cswdtx_alexandre.py @@ -2,6 +2,7 @@ import os import cPickle from blocks.initialization import IsotropicGaussian, Constant +from blocks.algorithms import Momentum import data from model.dest_mlp_tgtcls import Model, Stream @@ -29,8 +30,8 @@ embed_weights_init = IsotropicGaussian(0.01) mlp_weights_init = IsotropicGaussian(0.1) mlp_biases_init = Constant(0.01) -learning_rate = 0.01 -momentum = 0.9 +step_rule = Momentum(learning_rate=0.01, momentum=0.9) + batch_size = 200 valid_set = 'cuts/test_times_0' |