linfan 1 year ago
parent
commit
78c916a01c
2 changed files with 14 additions and 10 deletions
  1. 11 9
      recommend.py
  2. 3 1
      video_recall.py

+ 11 - 9
recommend.py

@@ -198,15 +198,16 @@ def video_recommend(request_id, mid, uid, size, top_K, flow_pool_P, app_type, al
 
     # load test config
     exp_config = None
-    if ab_code == 60058 or ab_code == 60059 or ab_code == 60060 or ab_code == 60061:
-        if ab_code == 60058:
-            exp_config = pool_recall.get_hotrecall_config()
-        elif ab_code == 60059:
-            exp_config = pool_recall.get_w2v_config()
-        elif ab_code == 60060:
-            exp_config = pool_recall.get_test_config()
-        elif ab_code == 60061:
-            exp_config = pool_recall.get_simrecall_config()
+
+    if ab_code == 60058:
+        exp_config = pool_recall.get_hotrecall_config()
+    elif ab_code == 60059:
+        exp_config = pool_recall.get_w2v_config()
+    elif ab_code == 60060:
+        exp_config = pool_recall.get_test_config()
+    elif ab_code == 60061:
+        exp_config = pool_recall.get_simrecall_config()
+    print("exp_config:", exp_config)
     if app_type in [config_.APP_TYPE['LAO_HAO_KAN_VIDEO'], config_.APP_TYPE['ZUI_JING_QI']]:
         t = [gevent.spawn(pool_recall.rov_pool_recall_with_region, size, expire_time, ab_code, exp_config)]
         if ab_code==60058:
@@ -439,6 +440,7 @@ def video_old_recommend(request_id, mid, uid, size, top_K, flow_pool_P, app_type
         exp_config = pool_recall.get_flow_exp_8_config()
     elif ab_code == 60057:
         exp_config = pool_recall.get_flow_exp_6_config()
+    print("exp_config:", exp_config)
 
     if app_type in [config_.APP_TYPE['LAO_HAO_KAN_VIDEO'], config_.APP_TYPE['ZUI_JING_QI']]:
         t = [gevent.spawn(pool_recall.rov_pool_recall_with_region, size, expire_time, ab_code, exp_config)]

+ 3 - 1
video_recall.py

@@ -1276,7 +1276,9 @@ class PoolRecall(object):
         recall_result = []
         recall_num = size
         if ab_code and exp_config:
-            if ab_code==60058 or ab_code==60059 or ab_code == 60060 or ab_code == 60061 :
+            if ab_code==60058 or ab_code==60059 or ab_code == 60060 \
+                    or ab_code == 60061 or ab_code==60052 \
+                    or ab_code==60053 or ab_code==60057 :
                 try:
                     recall_num = int(exp_config['recall_num'])
                 except: