|
@@ -35,6 +35,7 @@ class StaticModel():
|
|
|
|
|
|
# 第一层的输入维度是所有特征的embedding拼接
|
|
|
self.input_dim = self.tag_feature_num + self.stat_features_num*3
|
|
|
+ print(f"debug debug data input dim: {self.input_dim}")
|
|
|
|
|
|
|
|
|
def _init_hyper_parameters(self):
|
|
@@ -95,7 +96,7 @@ class StaticModel():
|
|
|
return fetch_dict
|
|
|
else:
|
|
|
label,left_features, right_features = input
|
|
|
- print(f"Label shape: {label.shape}")
|
|
|
+
|
|
|
# 获取相似度和特征向量
|
|
|
sim_score, left_vec, right_vec = dssm_model(left_features, right_features)
|
|
|
|