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/time_simple_mlp_tgtcls_2_cswdtx.py | |
| parent | 39e549f05e568e4153381f025b3a0f256e9a7b7a (diff) | |
| download | taxi-1137abf0511359df8f468f1d9a4828f3e0cb1b6a.tar.gz taxi-1137abf0511359df8f468f1d9a4828f3e0cb1b6a.zip | |
Fix embeddings sizes.
Diffstat (limited to 'config/time_simple_mlp_tgtcls_2_cswdtx.py')
| -rw-r--r-- | config/time_simple_mlp_tgtcls_2_cswdtx.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/config/time_simple_mlp_tgtcls_2_cswdtx.py b/config/time_simple_mlp_tgtcls_2_cswdtx.py index d252e38..608ed7e 100644 --- a/config/time_simple_mlp_tgtcls_2_cswdtx.py +++ b/config/time_simple_mlp_tgtcls_2_cswdtx.py @@ -14,8 +14,8 @@ for i in range(22): tgtcls.append(tgtcls[-1] + tgtcls[-2]) dim_embeddings = [ - ('origin_call', data.origin_call_size+1, 10), - ('origin_stand', data.stands_size+1, 10), + ('origin_call', data.origin_call_size, 10), + ('origin_stand', data.stands_size, 10), ('week_of_year', 52, 10), ('day_of_week', 7, 10), ('qhour_of_day', 24 * 4, 10), |