diff options
Diffstat (limited to 'config')
| -rw-r--r-- | config/bidirectional_1.py | 9 | ||||
| -rw-r--r-- | config/bidirectional_tgtcls_1.py | 10 |
2 files changed, 7 insertions, 12 deletions
diff --git a/config/bidirectional_1.py b/config/bidirectional_1.py index 35039d2..6970ece 100644 --- a/config/bidirectional_1.py +++ b/config/bidirectional_1.py @@ -10,7 +10,7 @@ dim_embeddings = [ ('week_of_year', 52, 10), ('day_of_week', 7, 10), ('qhour_of_day', 24 * 4, 10), - ('taxi_id', 448, 10), + ('taxi_id', data.taxi_id_size, 10), ] hidden_state_dim = 100 @@ -18,11 +18,8 @@ hidden_state_dim = 100 dim_hidden = [500, 500] embed_weights_init = IsotropicGaussian(0.01) -fork_weights_init = IsotropicGaussian(0.1) -fork_biases_init = Constant(0.01) -rec_weights_init = IsotropicGaussian(0.1) -mlp_weights_init = IsotropicGaussian(0.1) -mlp_biases_init = Constant(0.01) +weights_init = IsotropicGaussian(0.1) +biases_init = Constant(0.01) batch_size = 20 batch_sort_size = 20 diff --git a/config/bidirectional_tgtcls_1.py b/config/bidirectional_tgtcls_1.py index 88328e4..673c1e4 100644 --- a/config/bidirectional_tgtcls_1.py +++ b/config/bidirectional_tgtcls_1.py @@ -15,7 +15,7 @@ dim_embeddings = [ ('week_of_year', 52, 10), ('day_of_week', 7, 10), ('qhour_of_day', 24 * 4, 10), - ('taxi_id', 448, 10), + ('taxi_id', data.taxi_id_size, 10), ] hidden_state_dim = 100 @@ -23,13 +23,11 @@ hidden_state_dim = 100 dim_hidden = [500, 500] embed_weights_init = IsotropicGaussian(0.01) -fork_weights_init = IsotropicGaussian(0.1) -fork_biases_init = Constant(0.01) -rec_weights_init = IsotropicGaussian(0.1) -mlp_weights_init = IsotropicGaussian(0.1) -mlp_biases_init = Constant(0.01) +weights_init = IsotropicGaussian(0.1) +biases_init = Constant(0.01) batch_size = 20 batch_sort_size = 20 max_splits = 100 + |