try timer

src/models/esmm/model.py

@@ -114,13 +114,19 @@ def model_predict2(device_id, diary_ids, device_dict, diary_dict, predict_fn):
         tmp.update(device_info)
         tmp.update(diary_info)
         features = {}
-        for (col, value) in tmp.items():
-            if col in int_columns:
-                features[col] = _int64_feature(int(value))
-            elif col in float_columns:
-                features[col] = _float_feature(float(value))
-            elif col in str_columns:
-                features[col] = _bytes_feature(str(value).encode(encoding="utf-8"))
+        # for (col, value) in tmp.items():
+        #     if col in int_columns:
+        #         features[col] = _int64_feature(int(value))
+        #     elif col in float_columns:
+        #         features[col] = _float_feature(float(value))
+        #     elif col in str_columns:
+        #         features[col] = _bytes_feature(str(value).encode(encoding="utf-8"))
+        for col in int_columns:
+            features[col] = _int64_feature(int(tmp[col]))
+        for col in float_columns:
+            features[col] = _float_feature(float(tmp[col]))
+        for col in str_columns:
+            features[col] = _bytes_feature(str(tmp[col]).encode(encoding="utf-8"))
         example = tf.train.Example(features=tf.train.Features(feature=features))
         examples.append(example.SerializeToString())
     total_1 = (timeit.default_timer() - time_1)