|
@@ -69,9 +69,9 @@ class StaticModel():
|
|
|
# 获取相似度和特征向量
|
|
|
sim_score, left_vec, right_vec = dssm_model(left_features, right_features)
|
|
|
|
|
|
- self.inference_target_var = sim_score
|
|
|
- self.left_vector = left_vec
|
|
|
- self.right_vector = right_vec
|
|
|
+ self.inference_target_var = left_vec
|
|
|
+ # self.left_vector = left_vec
|
|
|
+ # self.right_vector = right_vec
|
|
|
|
|
|
# 计算损失
|
|
|
# 使用带margin的二元交叉熵损失
|