diff options
| author | Alex Auvolat <alex.auvolat@ens.fr> | 2015-05-04 17:27:56 -0400 |
|---|---|---|
| committer | Alex Auvolat <alex.auvolat@ens.fr> | 2015-05-04 17:27:56 -0400 |
| commit | 1556e9087f7e49bd75c8e236d2d3fb4fd936dc40 (patch) | |
| tree | aa97cc78177ba28d93c7a0d3ba9e9d8fc49d3acb /config/simple_mlp_tgtcls_0.py | |
| parent | 5f42c01231ccec377196472b6f4682b6afeb878d (diff) | |
| download | taxi-1556e9087f7e49bd75c8e236d2d3fb4fd936dc40.tar.gz taxi-1556e9087f7e49bd75c8e236d2d3fb4fd936dc40.zip | |
Fix it
Diffstat (limited to 'config/simple_mlp_tgtcls_0.py')
| -rw-r--r-- | config/simple_mlp_tgtcls_0.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/config/simple_mlp_tgtcls_0.py b/config/simple_mlp_tgtcls_0.py index 9ccfbfd..91770c2 100644 --- a/config/simple_mlp_tgtcls_0.py +++ b/config/simple_mlp_tgtcls_0.py @@ -17,7 +17,7 @@ with open(data.DATA_PATH + "/arrival-clusters.pkl") as f: tgtcls = cPickle.load( dim_embed = 10 dim_input = n_begin_end_pts * 2 * 2 + dim_embed + dim_embed -dim_hidden = [200] +dim_hidden = [] dim_output = tgtcls.shape[0] learning_rate = 0.0001 |