update defun

src/main_portrait.py

@@ -5,7 +5,8 @@ import time
 
 import tensorflow as tf
 
-from models.esmm.fe import (get_device_dict_from_redis, get_diary_dict_from_redis)
+from models.esmm import device_fe as device_fe
+from models.esmm import diary_fe as diary_fe
 from models.esmm.model import model_predict_diary
 from utils.cache import redis_client2
 from utils.grey import recommed_service_category_device_id_by_tail
@@ -56,8 +57,8 @@ def offline_predict(device_id, device_dict, diary_dict, predict_fn):
 def main():
     tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.INFO)
 
-    device_dict = get_device_dict_from_redis()
-    diary_dict = get_diary_dict_from_redis()
+    device_dict = device_fe.get_device_dict_from_redis()
+    diary_dict = diary_fe.get_diary_dict_from_redis()
     print("redis data: " + str(len(device_dict)) + " " + str(len(diary_dict)))
 
     save_path = "/home/gmuser/data/models/1596018742"