|
@@ -50,9 +50,9 @@ class StaticModel():
|
|
|
)
|
|
|
|
|
|
if is_infer:
|
|
|
- sample_id, left_features, right_features = input
|
|
|
+ left_features, right_features = input
|
|
|
else:
|
|
|
- sample_id, left_features, right_features, label = input
|
|
|
+ left_features, right_features, label = input
|
|
|
|
|
|
# 获取相似度和特征向量
|
|
|
sim_score, left_vec, right_vec = dssm_model(left_features, right_features)
|