add prediction

src/models/esmm/model.py

@@ -95,6 +95,7 @@ def _bytes_feature(value):
 
 
 def model_predict_diary(device_id, diary_ids, device_dict, diary_dict, predict_fn):
+    try:
         time_1 = timeit.default_timer()
         device_info, diary_lst, diary_ids_res = device_diary_fe(device_id, diary_ids, device_dict, diary_dict)
         print("predict check: " + str(len(diary_lst)) + " " + str(len(diary_ids_res)))
@@ -137,3 +138,6 @@ def model_predict_diary(device_id, diary_ids, device_dict, diary_dict, predict_f
         total_1 = (timeit.default_timer() - time_1)
         print("prediction cost {:.5f}s".format(total_1))
         return res
+    except Exception as e:
+        print(e)
+        return []