Skip to content

Commit

Permalink
update dssm simlary gamma parameter (#61)
Browse files Browse the repository at this point in the history
  • Loading branch information
zzszmyf authored Jan 16, 2022
1 parent 106bb4e commit a8a84b6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion deepmatch/models/dssm.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,7 @@ def DSSM(user_feature_columns, item_feature_columns, user_dnn_hidden_units=(64,
item_dnn_out = DNN(item_dnn_hidden_units, dnn_activation, l2_reg_dnn, dnn_dropout,
dnn_use_bn, seed=seed)(item_dnn_input)

score = Similarity(type=metric)([user_dnn_out, item_dnn_out])
score = Similarity(type=metric, gamma = 10)([user_dnn_out, item_dnn_out])

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

Expand Down

0 comments on commit a8a84b6

Please sign in to comment.