Skip to content

Commit a8a84b6

Browse files
authored
update dssm simlary gamma parameter (#61)
1 parent 106bb4e commit a8a84b6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

deepmatch/models/dssm.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def DSSM(user_feature_columns, item_feature_columns, user_dnn_hidden_units=(64,
6060
item_dnn_out = DNN(item_dnn_hidden_units, dnn_activation, l2_reg_dnn, dnn_dropout,
6161
dnn_use_bn, seed=seed)(item_dnn_input)
6262

63-
score = Similarity(type=metric)([user_dnn_out, item_dnn_out])
63+
score = Similarity(type=metric, gamma = 10)([user_dnn_out, item_dnn_out])
6464

6565
output = PredictionLayer("binary", False)(score)
6666

0 commit comments

Comments
 (0)