dssm train

recommend-model-produce/src/main/python/tools/inferv2.py

@@ -53,7 +53,6 @@ def main():
         input_data[vid] = left_features
         
 
-    input_data2 = {k: input_data[k] for k in list(input_data)[:5]}
     # 设置输入
 
     result = {}
@@ -61,7 +60,7 @@ def main():
     with open('/app/data.json', 'w') as json_file:
         json_file.write("")
 
-    for k,v in input_data2.items():
+    for k,v in input_data.items():
         v2 = np.array([v], dtype=np.float32)
         input_handle.copy_from_cpu(v2)
         # 运行predictor