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_2_cswdtx.py | |
| parent | 39e549f05e568e4153381f025b3a0f256e9a7b7a (diff) | |
| download | taxi-1137abf0511359df8f468f1d9a4828f3e0cb1b6a.tar.gz taxi-1137abf0511359df8f468f1d9a4828f3e0cb1b6a.zip | |
Fix embeddings sizes.
Diffstat (limited to 'config/time_simple_mlp_2_cswdtx.py')
| -rw-r--r-- | config/time_simple_mlp_2_cswdtx.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/config/time_simple_mlp_2_cswdtx.py b/config/time_simple_mlp_2_cswdtx.py index 0747e27..2ec28c6 100644 --- a/config/time_simple_mlp_2_cswdtx.py +++ b/config/time_simple_mlp_2_cswdtx.py @@ -9,8 +9,8 @@ n_begin_end_pts = 5 # how many points we consider at the beginning and end o n_valid = 1000 dim_embeddings = [ - ('origin_call', data.n_train_clients+1, 10), - ('origin_stand', data.n_stands+1, 10), + ('origin_call', data.origin_call_train_size, 10), + ('origin_stand', data.stands_size, 10), ('week_of_year', 52, 10), ('day_of_week', 7, 10), ('qhour_of_day', 24 * 4, 10), |