get result

src/models/esmm/model.py

@@ -129,8 +129,11 @@ def model_predict_diary(device_id, diary_ids, device_dict, diary_dict, predict_f
 
     time_1 = timeit.default_timer()
     predictions = predict_fn({"examples": examples})
-    res = sorted(zip(diary_ids_res, predictions["output"].tolist()), key=lambda x: x[1], reverse=True)
+    res_tuple = sorted(zip(diary_ids_res, predictions["output"].tolist()), key=lambda x: x[1], reverse=True)
+    res = []
+    for (id, _) in res_tuple:
+        res.append(id)
     print(res)
     total_1 = (timeit.default_timer() - time_1)
     print("prediction cost {:.5f}s".format(total_1))
-    return predictions
+    return res