update type

src/models/esmm/model.py

@@ -133,8 +133,7 @@ def model_predict2(device_id, diary_ids, device_dict, diary_dict, predict_fn):
     for i in diary_lst:
         diary_ids.append(i.get("card_id", "-1"))
     print(predictions)
-    print(predictions["output"])
-    print(predictions["output"].values)
+    print(predictions["output"].tolist())
     print(diary_ids)
     total_1 = (timeit.default_timer() - time_1)
     print("prediction cost {:.5f}s".format(total_1))