linfan пре 1 година
родитељ
комит
3150068494
2 измењених фајлова са 2 додато и 1 уклоњено
  1. 1 0
      config.py
  2. 1 1
      video_recall.py

+ 1 - 0
config.py

@@ -470,6 +470,7 @@ class BaseConfig(object):
         'hot_3_day_recall': 'hot_3_day_recall',  # 3天召回
         'hot_recall': 'hot_recall',  # hot召回
         'w2v_recall': 'w2v_recall', #word2vec
+        'u2i_tag_recall':'u2i_tag_recall', #u2i_tag_recall
     }
 
     # category id mapping

+ 1 - 1
video_recall.py

@@ -2685,7 +2685,7 @@ class PoolRecall(object):
                     vid = int(per_item[0])
                     video_ids.append(vid)
                     recall_dict[vid] = {'videoId': vid, 'flowPool': '',
-                         'rovScore': per_item[1], 'pushFrom': config_.PUSH_FROM['u2i_tag_recall'],
+                         'rovScore': float(per_item[1]), 'pushFrom': config_.PUSH_FROM['u2i_tag_recall'],
                          'abCode': self.ab_code}
                 except Exception as e:
                     continue