浏览代码

save model

丁云鹏 5 月之前
父节点
当前提交
7cee06b088
共有 1 个文件被更改,包括 3 次插入3 次删除
  1. 3 3
      recommend-model-produce/src/main/python/models/dssm/static_model.py

+ 3 - 3
recommend-model-produce/src/main/python/models/dssm/static_model.py

@@ -69,9 +69,9 @@ class StaticModel():
             # 获取相似度和特征向量
             # 获取相似度和特征向量
             sim_score, left_vec, right_vec = dssm_model(left_features, right_features)
             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的二元交叉熵损失
             # 使用带margin的二元交叉熵损失