diff options
| author | Étienne Simon <esimon@esimon.eu> | 2015-05-22 11:23:49 -0400 |
|---|---|---|
| committer | Étienne Simon <esimon@esimon.eu> | 2015-05-22 11:24:50 -0400 |
| commit | 1137abf0511359df8f468f1d9a4828f3e0cb1b6a (patch) | |
| tree | 76b0336d5ed3ee835b0efb902ab06ef4543a854d /config/joint_simple_mlp_tgtcls_111_cswdtx_bigger_dropout.py | |
| parent | 39e549f05e568e4153381f025b3a0f256e9a7b7a (diff) | |
| download | taxi-1137abf0511359df8f468f1d9a4828f3e0cb1b6a.tar.gz taxi-1137abf0511359df8f468f1d9a4828f3e0cb1b6a.zip | |
Fix embeddings sizes.
Diffstat (limited to 'config/joint_simple_mlp_tgtcls_111_cswdtx_bigger_dropout.py')
| -rw-r--r-- | config/joint_simple_mlp_tgtcls_111_cswdtx_bigger_dropout.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/config/joint_simple_mlp_tgtcls_111_cswdtx_bigger_dropout.py b/config/joint_simple_mlp_tgtcls_111_cswdtx_bigger_dropout.py index a6af132..937122d 100644 --- a/config/joint_simple_mlp_tgtcls_111_cswdtx_bigger_dropout.py +++ b/config/joint_simple_mlp_tgtcls_111_cswdtx_bigger_dropout.py @@ -20,8 +20,8 @@ for i in range(21): time_tgtcls.append(time_tgtcls[-1] + time_tgtcls[-2]) dim_embeddings = [ - ('origin_call', data.origin_call_size+1, 15), - ('origin_stand', data.stands_size+1, 10), + ('origin_call', data.origin_call_size, 15), + ('origin_stand', data.stands_size, 10), ('week_of_year', 52, 10), ('day_of_week', 7, 10), ('qhour_of_day', 24 * 4, 10), |