get result

src/models/esmm/model.py

@@ -124,24 +124,13 @@ def model_predict_diary(device_id, diary_ids, device_dict, diary_dict, predict_f
         example = tf.train.Example(features=tf.train.Features(feature=features))
         examples.append(example.SerializeToString())
 
-    print(examples)
-    a = json.dumps(examples)
-    print(a)
-    b = json.loads(examples)
     total_1 = (timeit.default_timer() - time_1)
     print("make example cost {:.5f}s".format(total_1))
 
     time_1 = timeit.default_timer()
-    # predictions = predict_fn({"examples": examples})
-    predictions = predict_fn({"examples": b})
-    diary_ids = []
-    for i in diary_lst:
-        diary_ids.append(i.get("card_id", "-1"))
-    print(predictions)
-    print(predictions["output"].tolist())
-    print(diary_ids_res)
-    print(diary_ids)
-    print(len(predictions["output"].tolist()), len(diary_ids), len(diary_ids_res))
+    predictions = predict_fn({"examples": examples})
+    res = sorted(zip(diary_ids_res, predictions["output"].tolist()), key=lambda x: x[1], reverse=True)
+    print(res)
     total_1 = (timeit.default_timer() - time_1)
     print("prediction cost {:.5f}s".format(total_1))
     return predictions