diff options
Diffstat (limited to 'config')
| -rw-r--r-- | config/joint_simple_mlp_tgtcls_1_cswdtx.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/config/joint_simple_mlp_tgtcls_1_cswdtx.py b/config/joint_simple_mlp_tgtcls_1_cswdtx.py index f3de40b..74b3c75 100644 --- a/config/joint_simple_mlp_tgtcls_1_cswdtx.py +++ b/config/joint_simple_mlp_tgtcls_1_cswdtx.py @@ -41,6 +41,9 @@ dim_output_dest = len(dest_tgtcls) dim_hidden_time = [] dim_output_time = len(time_tgtcls) +# Cost ratio between distance cost and time cost +time_cost_factor = 4 + embed_weights_init = IsotropicGaussian(0.001) mlp_weights_init = IsotropicGaussian(0.01) mlp_biases_init = Constant(0.001) |