video_rank.py 50 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142
  1. import json
  2. import random
  3. import numpy
  4. from log import Log
  5. from config import set_config
  6. from video_recall import PoolRecall
  7. from db_helper import RedisHelper
  8. from utils import FilterVideos, send_msg_to_feishu
  9. from rank_service import get_featurs, get_tf_serving_sores
  10. log_ = Log()
  11. config_ = set_config()
  12. def video_rank(data, size, top_K, flow_pool_P):
  13. """
  14. 视频分发排序
  15. :param data: 各路召回的视频 type-dict {'rov_pool_recall': [], 'flow_pool_recall': []}
  16. :param size: 请求数
  17. :param top_K: 保证topK为召回池视频 type-int
  18. :param flow_pool_P: size-top_K视频为流量池视频的概率 type-float
  19. :return: rank_result
  20. """
  21. if not data['rov_pool_recall'] and not data['flow_pool_recall']:
  22. return []
  23. # 将各路召回的视频按照score从大到小排序
  24. # 最惊奇相关推荐相似视频
  25. # relevant_recall = [item for item in data['rov_pool_recall']
  26. # if item.get('pushFrom') == config_.PUSH_FROM['top_video_relevant_appType_19']]
  27. # relevant_recall_rank = sorted(relevant_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  28. # 最惊奇完整影视视频
  29. # whole_movies_recall = [item for item in data['rov_pool_recall']
  30. # if item.get('pushFrom') == config_.PUSH_FROM['whole_movies']]
  31. # whole_movies_recall_rank = sorted(whole_movies_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  32. # 最惊奇影视解说视频
  33. # talk_videos_recall = [item for item in data['rov_pool_recall']
  34. # if item.get('pushFrom') == config_.PUSH_FROM['talk_videos']]
  35. # talk_videos_recall_rank = sorted(talk_videos_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  36. # 小时级更新数据
  37. # h_recall = [item for item in data['rov_pool_recall'] if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_h']]
  38. # h_recall_rank = sorted(h_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  39. # 相对30天天级规则更新数据
  40. day_30_recall = [item for item in data['rov_pool_recall']
  41. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_30day']]
  42. day_30_recall_rank = sorted(day_30_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  43. # 地域分组小时级规则更新数据
  44. region_h_recall = [item for item in data['rov_pool_recall']
  45. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_region_h']]
  46. region_h_recall_rank = sorted(region_h_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  47. # 地域分组小时级更新24h规则更新数据
  48. region_24h_recall = [item for item in data['rov_pool_recall']
  49. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_region_24h']]
  50. region_24h_recall_rank = sorted(region_24h_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  51. # 地域分组天级规则更新数据
  52. # region_day_recall = [item for item in data['rov_pool_recall']
  53. # if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_region_day']]
  54. # region_day_recall_rank = sorted(region_day_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  55. # 相对24h规则更新数据
  56. rule_24h_recall = [item for item in data['rov_pool_recall']
  57. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_24h']]
  58. rule_24h_recall_rank = sorted(rule_24h_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  59. # 相对24h规则筛选后剩余更新数据
  60. rule_24h_dup_recall = [item for item in data['rov_pool_recall']
  61. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_24h_dup']]
  62. rule_24h_dup_recall_rank = sorted(rule_24h_dup_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  63. # 相对48h规则更新数据
  64. rule_48h_recall = [item for item in data['rov_pool_recall']
  65. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_48h']]
  66. rule_48h_recall_rank = sorted(rule_48h_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  67. # 相对48h规则筛选后剩余更新数据
  68. rule_48h_dup_recall = [item for item in data['rov_pool_recall']
  69. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_48h_dup']]
  70. rule_48h_dup_recall_rank = sorted(rule_48h_dup_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  71. # 天级规则更新数据
  72. # day_recall = [item for item in data['rov_pool_recall'] if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_day']]
  73. # day_recall_rank = sorted(day_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  74. # ROV召回池
  75. # rov_initial_recall = [
  76. # item for item in data['rov_pool_recall']
  77. # if item.get('pushFrom') not in
  78. # [config_.PUSH_FROM['top_video_relevant_appType_19'],
  79. # config_.PUSH_FROM['rov_recall_h'],
  80. # config_.PUSH_FROM['rov_recall_region_h'],
  81. # config_.PUSH_FROM['rov_recall_region_24h'],
  82. # config_.PUSH_FROM['rov_recall_region_day'],
  83. # config_.PUSH_FROM['rov_recall_24h'],
  84. # config_.PUSH_FROM['rov_recall_24h_dup'],
  85. # config_.PUSH_FROM['rov_recall_48h'],
  86. # config_.PUSH_FROM['rov_recall_48h_dup'],
  87. # config_.PUSH_FROM['rov_recall_day'],
  88. # config_.PUSH_FROM['whole_movies'],
  89. # config_.PUSH_FROM['talk_videos']]
  90. # ]
  91. # rov_initial_recall_rank = sorted(rov_initial_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  92. # rov_recall_rank = whole_movies_recall_rank + talk_videos_recall_rank + h_recall_rank + \
  93. # day_30_recall_rank + region_h_recall_rank + region_24h_recall_rank + \
  94. # region_day_recall_rank + rule_24h_recall_rank + rule_24h_dup_recall_rank + \
  95. # rule_48h_recall_rank + rule_48h_dup_recall_rank + \
  96. # day_recall_rank + rov_initial_recall_rank
  97. rov_recall_rank = day_30_recall_rank + \
  98. region_h_recall_rank + region_24h_recall_rank + \
  99. rule_24h_recall_rank + rule_24h_dup_recall_rank + \
  100. rule_48h_recall_rank + rule_48h_dup_recall_rank
  101. # 流量池
  102. flow_recall_rank = sorted(data['flow_pool_recall'], key=lambda k: k.get('rovScore', 0), reverse=True)
  103. # 对各路召回的视频进行去重
  104. rov_recall_rank, flow_recall_rank = remove_duplicate(rov_recall=rov_recall_rank, flow_recall=flow_recall_rank,
  105. top_K=top_K)
  106. # log_.info('remove_duplicate finished! rov_recall_rank = {}, flow_recall_rank = {}'.format(
  107. # rov_recall_rank, flow_recall_rank))
  108. # rank_result = relevant_recall_rank
  109. rank_result = []
  110. # 从ROV召回池中获取top k
  111. if len(rov_recall_rank) > 0:
  112. rank_result.extend(rov_recall_rank[:top_K])
  113. rov_recall_rank = rov_recall_rank[top_K:]
  114. else:
  115. rank_result.extend(flow_recall_rank[:top_K])
  116. flow_recall_rank = flow_recall_rank[top_K:]
  117. # 按概率 p 及score排序获取 size - k 个视频
  118. i = 0
  119. while i < size - top_K:
  120. # 随机生成[0, 1)浮点数
  121. rand = random.random()
  122. # log_.info('rand: {}'.format(rand))
  123. if rand < flow_pool_P:
  124. if flow_recall_rank:
  125. rank_result.append(flow_recall_rank[0])
  126. flow_recall_rank.remove(flow_recall_rank[0])
  127. else:
  128. rank_result.extend(rov_recall_rank[:size - top_K - i])
  129. return rank_result[:size]
  130. else:
  131. if rov_recall_rank:
  132. rank_result.append(rov_recall_rank[0])
  133. rov_recall_rank.remove(rov_recall_rank[0])
  134. else:
  135. rank_result.extend(flow_recall_rank[:size - top_K - i])
  136. return rank_result[:size]
  137. i += 1
  138. return rank_result[:size]
  139. def video_new_rank(videoIds, fast_flow_set, flow_set, size, top_K, flow_pool_P):
  140. """
  141. 视频分发排序
  142. :param data: 各路召回的视频 type-dict {'rov_pool_recall': [], 'flow_pool_recall': []}
  143. :param size: 请求数
  144. :param top_K: 保证topK为召回池视频 type-int
  145. :param flow_pool_P: size-top_K视频为流量池视频的概率 type-float
  146. :return: rank_result
  147. """
  148. add_flow_set = set('')
  149. if not videoIds or len(videoIds)==0:
  150. return [], add_flow_set
  151. redisObj = RedisHelper()
  152. vidKeys = []
  153. for vid in videoIds:
  154. vidKeys.append("k_p:"+str(vid))
  155. #print("vidKeys:", vidKeys)
  156. video_scores = redisObj.get_batch_key(vidKeys)
  157. #print(video_scores)
  158. video_items = []
  159. for i in range(len(video_scores)):
  160. try:
  161. #print(video_scores[i])
  162. if video_scores[i] is None:
  163. video_items.append((videoIds[i], 0.0))
  164. else:
  165. video_score_str = json.loads(video_scores[i])
  166. #print("video_score_str:",video_score_str)
  167. video_items.append((videoIds[i], video_score_str[0]))
  168. except Exception:
  169. video_items.append((videoIds[i], 0.0))
  170. sort_items = sorted(video_items, key=lambda k: k[1], reverse=True)
  171. #print("sort_items:", sort_items)
  172. rov_recall_rank = sort_items
  173. fast_flow_recall_rank = []
  174. flow_recall_rank = []
  175. for item in sort_items:
  176. if item[0] in fast_flow_set:
  177. fast_flow_recall_rank.append(item)
  178. elif item[0] in flow_set:
  179. flow_recall_rank.append(item)
  180. # all flow result
  181. all_flow_recall_rank = fast_flow_recall_rank+flow_recall_rank
  182. rank_result = []
  183. rank_set = set('')
  184. # 从ROV召回池中获取top k
  185. if len(rov_recall_rank) > 0:
  186. rank_result.extend(rov_recall_rank[:top_K])
  187. rov_recall_rank = rov_recall_rank[top_K:]
  188. else:
  189. rank_result.extend(all_flow_recall_rank[:top_K])
  190. all_flow_recall_rank = all_flow_recall_rank[top_K:]
  191. for rank_item in rank_result:
  192. rank_set.add(rank_item[0])
  193. #print("rank_result:", rank_result)
  194. # 按概率 p 及score排序获取 size - k 个视频, 第4个位置按概率取流量池
  195. i = 0
  196. left_quato = size - top_K
  197. j = 0
  198. jj = 0
  199. while i < left_quato and (j<len(all_flow_recall_rank) or jj<len(rov_recall_rank)):
  200. # 随机生成[0, 1)浮点数
  201. rand = random.random()
  202. # log_.info('rand: {}'.format(rand))
  203. if rand < flow_pool_P:
  204. for flow_item in all_flow_recall_rank:
  205. j+=1
  206. if flow_item[0] in rank_set:
  207. continue
  208. else:
  209. rank_result.append(flow_item)
  210. rank_set.add(flow_item[0])
  211. add_flow_set.add(flow_item[0])
  212. i += 1
  213. if i>= left_quato:
  214. break
  215. else:
  216. for recall_item in rov_recall_rank:
  217. jj+=1
  218. if recall_item[0] in rank_set:
  219. continue
  220. else:
  221. rank_result.append(recall_item)
  222. rank_set.add(recall_item[0])
  223. i += 1
  224. if i>= left_quato:
  225. break
  226. #print("rank_result:", rank_result)
  227. #print("add_flow_set:", add_flow_set)
  228. return rank_result[:size], add_flow_set
  229. def refactor_video_rank(rov_recall_rank, fast_flow_set, flow_set, size, top_K, flow_pool_P):
  230. """
  231. 视频分发排序
  232. :param data: 各路召回的视频 type-dict {'rov_pool_recall': [], 'flow_pool_recall': []}
  233. :param size: 请求数
  234. :param top_K: 保证topK为召回池视频 type-int
  235. :param flow_pool_P: size-top_K视频为流量池视频的概率 type-float
  236. :return: rank_result
  237. """
  238. if not rov_recall_rank or len(rov_recall_rank) == 0:
  239. return []
  240. fast_flow_recall_rank = []
  241. flow_recall_rank = []
  242. for item in rov_recall_rank:
  243. vid = item.get('videoId', 0)
  244. #print(item)
  245. if vid in fast_flow_set:
  246. fast_flow_recall_rank.append(item)
  247. elif vid in flow_set:
  248. flow_recall_rank.append(item)
  249. # all flow result
  250. all_flow_recall_rank = fast_flow_recall_rank + flow_recall_rank
  251. rank_result = []
  252. rank_set = set('')
  253. # 从ROV召回池中获取top k
  254. if len(rov_recall_rank) > 0:
  255. rank_result.extend(rov_recall_rank[:top_K])
  256. rov_recall_rank = rov_recall_rank[top_K:]
  257. else:
  258. rank_result.extend(all_flow_recall_rank[:top_K])
  259. all_flow_recall_rank = all_flow_recall_rank[top_K:]
  260. #已存放了多少VID
  261. for rank_item in rank_result:
  262. rank_set.add(rank_item.get('videoId', 0))
  263. # 按概率 p 及score排序获取 size - k 个视频, 第4个位置按概率取流量池
  264. i = 0
  265. while i < size - top_K:
  266. # 随机生成[0, 1)浮点数
  267. rand = random.random()
  268. # log_.info('rand: {}'.format(rand))
  269. if rand < flow_pool_P:
  270. for flow_item in all_flow_recall_rank:
  271. flow_vid = flow_item.get('videoId', 0)
  272. if flow_vid in rank_set:
  273. continue
  274. else:
  275. rank_result.append(flow_item)
  276. rank_set.add(flow_vid)
  277. else:
  278. for recall_item in rov_recall_rank:
  279. flow_vid = recall_item.get('videoId', 0)
  280. if flow_vid in rank_set:
  281. continue
  282. else:
  283. rank_result.append(recall_item)
  284. rank_set.add(flow_vid)
  285. i += 1
  286. return rank_result[:size]
  287. def remove_duplicate(rov_recall, flow_recall, top_K):
  288. """
  289. 对多路召回的视频去重
  290. 去重原则:
  291. 如果视频在ROV召回池topK,则保留ROV召回池,否则保留流量池
  292. :param rov_recall: ROV召回池-已排序
  293. :param flow_recall: 流量池-已排序
  294. :param top_K: 保证topK为召回池视频 type-int
  295. :return:
  296. """
  297. flow_recall_result = []
  298. rov_recall_remove = []
  299. flow_recall_video_ids = [item['videoId'] for item in flow_recall]
  300. # rov_recall topK
  301. for item in rov_recall[:top_K]:
  302. if item['videoId'] in flow_recall_video_ids:
  303. flow_recall_video_ids.remove(item['videoId'])
  304. # other
  305. for item in rov_recall[top_K:]:
  306. if item['videoId'] in flow_recall_video_ids:
  307. rov_recall_remove.append(item)
  308. # rov recall remove
  309. for item in rov_recall_remove:
  310. rov_recall.remove(item)
  311. # flow recall remove
  312. for item in flow_recall:
  313. if item['videoId'] in flow_recall_video_ids:
  314. flow_recall_result.append(item)
  315. return rov_recall, flow_recall_result
  316. def bottom_strategy(request_id, size, app_type, ab_code, params):
  317. """
  318. 兜底策略: 从ROV召回池中获取top1000,进行状态过滤后的视频
  319. :param request_id: request_id
  320. :param size: 需要获取的视频数
  321. :param app_type: 产品标识 type-int
  322. :param ab_code: abCode
  323. :param params:
  324. :return:
  325. """
  326. pool_recall = PoolRecall(request_id=request_id, app_type=app_type, ab_code=ab_code)
  327. key_name, _ = pool_recall.get_pool_redis_key(pool_type='rov')
  328. redis_helper = RedisHelper(params=params)
  329. data = redis_helper.get_data_zset_with_index(key_name=key_name, start=0, end=1000)
  330. if not data:
  331. log_.info('{} —— ROV推荐进入了二次兜底, data = {}'.format(config_.ENV_TEXT, data))
  332. send_msg_to_feishu('{} —— ROV推荐进入了二次兜底,请查看是否有数据更新失败问题。'.format(config_.ENV_TEXT))
  333. # 二次兜底
  334. bottom_data = bottom_strategy_last(size=size, app_type=app_type, ab_code=ab_code, params=params)
  335. return bottom_data
  336. # 视频状态过滤采用离线定时过滤方案
  337. # 状态过滤
  338. # filter_videos = FilterVideos(app_type=app_type, video_ids=data)
  339. # filtered_data = filter_videos.filter_video_status(video_ids=data)
  340. if len(data) > size:
  341. random_data = numpy.random.choice(data, size, False)
  342. else:
  343. random_data = data
  344. bottom_data = [{'videoId': int(item), 'pushFrom': config_.PUSH_FROM['bottom'], 'abCode': ab_code}
  345. for item in random_data]
  346. return bottom_data
  347. def bottom_strategy_last(size, app_type, ab_code, params):
  348. """
  349. 兜底策略: 从兜底视频中随机获取视频,进行状态过滤后的视频
  350. :param size: 需要获取的视频数
  351. :param app_type: 产品标识 type-int
  352. :param ab_code: abCode
  353. :param params:
  354. :return:
  355. """
  356. redis_helper = RedisHelper(params=params)
  357. bottom_data = redis_helper.get_data_zset_with_index(key_name=config_.BOTTOM_KEY_NAME, start=0, end=-1)
  358. random_data = numpy.random.choice(bottom_data, size * 30, False)
  359. # 视频状态过滤采用离线定时过滤方案
  360. # 状态过滤
  361. # filter_videos = FilterVideos(app_type=app_type, video_ids=random_data)
  362. # filtered_data = filter_videos.filter_video_status(video_ids=random_data)
  363. bottom_data = [{'videoId': int(video_id), 'pushFrom': config_.PUSH_FROM['bottom_last'], 'abCode': ab_code}
  364. for video_id in random_data[:size]]
  365. return bottom_data
  366. def bottom_strategy2(size, app_type, mid, uid, ab_code, client_info, params):
  367. """
  368. 兜底策略: 从兜底视频中随机获取视频,进行过滤后的视频
  369. :param size: 需要获取的视频数
  370. :param app_type: 产品标识 type-int
  371. :param mid: mid
  372. :param uid: uid
  373. :param ab_code: abCode
  374. :param client_info: 地域信息
  375. :param params:
  376. :return:
  377. """
  378. # 获取存在城市分组数据的城市编码列表
  379. city_code_list = [code for _, code in config_.CITY_CODE.items()]
  380. # 获取provinceCode
  381. province_code = client_info.get('provinceCode', '-1')
  382. # 获取cityCode
  383. city_code = client_info.get('cityCode', '-1')
  384. if city_code in city_code_list:
  385. # 分城市数据存在时,获取城市分组数据
  386. region_code = city_code
  387. else:
  388. region_code = province_code
  389. if region_code == '':
  390. region_code = '-1'
  391. redis_helper = RedisHelper(params=params)
  392. bottom_data = redis_helper.get_data_from_set(key_name=config_.BOTTOM2_KEY_NAME)
  393. bottom_result = []
  394. if bottom_data is None:
  395. return bottom_result
  396. if len(bottom_data) > 0:
  397. try:
  398. random_data = numpy.random.choice(bottom_data, size * 5, False)
  399. except Exception as e:
  400. random_data = bottom_data
  401. video_ids = [int(item) for item in random_data]
  402. # 过滤
  403. filter_ = FilterVideos(request_id=params.request_id, app_type=app_type, mid=mid, uid=uid, video_ids=video_ids)
  404. filtered_data = filter_.filter_videos(pool_type='flow', region_code=region_code)
  405. if filtered_data:
  406. bottom_result = [{'videoId': int(video_id), 'pushFrom': config_.PUSH_FROM['bottom2'], 'abCode': ab_code}
  407. for video_id in filtered_data[:size]]
  408. return bottom_result
  409. def video_rank_by_w_h_rate(videos):
  410. """
  411. 视频宽高比实验(每组的前两个视频调整为横屏视频),根据视频宽高比信息对视频进行重排
  412. :param videos:
  413. :return:
  414. """
  415. redis_helper = RedisHelper()
  416. # ##### 判断前两个视频是否是置顶视频 或者 流量池视频
  417. top_2_push_from_flag = [False, False]
  418. for i, video in enumerate(videos[:2]):
  419. if video['pushFrom'] in [config_.PUSH_FROM['top'], config_.PUSH_FROM['flow_recall']]:
  420. top_2_push_from_flag[i] = True
  421. if top_2_push_from_flag[0] and top_2_push_from_flag[1]:
  422. return videos
  423. # ##### 判断前两个视频是否为横屏
  424. top_2_w_h_rate_flag = [False, False]
  425. for i, video in enumerate(videos[:2]):
  426. if video['pushFrom'] in [config_.PUSH_FROM['top'], config_.PUSH_FROM['flow_recall']]:
  427. # 视频来源为置顶 或 流量池时,不做判断
  428. top_2_w_h_rate_flag[i] = True
  429. elif video['pushFrom'] in [config_.PUSH_FROM['rov_recall'], config_.PUSH_FROM['bottom']]:
  430. # 视频来源为 rov召回池 或 一层兜底时,判断是否是横屏
  431. w_h_rate = redis_helper.get_score_with_value(
  432. key_name=config_.W_H_RATE_UP_1_VIDEO_LIST_KEY_NAME['rov_recall'], value=video['videoId'])
  433. if w_h_rate is not None:
  434. top_2_w_h_rate_flag[i] = True
  435. elif video['pushFrom'] == config_.PUSH_FROM['bottom_last']:
  436. # 视频来源为 二层兜底时,判断是否是横屏
  437. w_h_rate = redis_helper.get_score_with_value(
  438. key_name=config_.W_H_RATE_UP_1_VIDEO_LIST_KEY_NAME['bottom_last'], value=video['videoId'])
  439. if w_h_rate is not None:
  440. top_2_w_h_rate_flag[i] = True
  441. if top_2_w_h_rate_flag[0] and top_2_w_h_rate_flag[1]:
  442. return videos
  443. # ##### 前两个视频中有不符合前面两者条件的,对视频进行位置调整
  444. # 记录横屏视频位置
  445. horizontal_video_index = []
  446. # 记录流量池视频位置
  447. flow_video_index = []
  448. # 记录置顶视频位置
  449. top_video_index = []
  450. for i, video in enumerate(videos):
  451. # 视频来源为置顶
  452. if video['pushFrom'] == config_.PUSH_FROM['top']:
  453. top_video_index.append(i)
  454. # 视频来源为流量池
  455. elif video['pushFrom'] == config_.PUSH_FROM['flow_recall']:
  456. flow_video_index.append(i)
  457. # 视频来源为rov召回池 或 一层兜底
  458. elif video['pushFrom'] in [config_.PUSH_FROM['rov_recall'], config_.PUSH_FROM['bottom']]:
  459. w_h_rate = redis_helper.get_score_with_value(
  460. key_name=config_.W_H_RATE_UP_1_VIDEO_LIST_KEY_NAME['rov_recall'], value=video['videoId'])
  461. if w_h_rate is not None:
  462. horizontal_video_index.append(i)
  463. else:
  464. continue
  465. # 视频来源为 二层兜底
  466. elif video['pushFrom'] == config_.PUSH_FROM['bottom_last']:
  467. w_h_rate = redis_helper.get_score_with_value(
  468. key_name=config_.W_H_RATE_UP_1_VIDEO_LIST_KEY_NAME['bottom_last'], value=video['videoId'])
  469. if w_h_rate is not None:
  470. horizontal_video_index.append(i)
  471. else:
  472. continue
  473. # 重新排序
  474. top2_index = []
  475. for i in range(2):
  476. if i in top_video_index:
  477. top2_index.append(i)
  478. elif i in flow_video_index:
  479. top2_index.append(i)
  480. flow_video_index.remove(i)
  481. elif i in horizontal_video_index:
  482. top2_index.append(i)
  483. horizontal_video_index.remove(i)
  484. elif len(horizontal_video_index) > 0:
  485. # 调整横屏视频到第一位
  486. top2_index.append(horizontal_video_index[0])
  487. # 从横屏位置记录中移除
  488. horizontal_video_index.pop(0)
  489. elif i == 0:
  490. return videos
  491. # 重排
  492. flow_result = [videos[i] for i in flow_video_index]
  493. other_result = [videos[i] for i in range(len(videos)) if i not in top2_index and i not in flow_video_index]
  494. top2_result = []
  495. for i, j in enumerate(top2_index):
  496. item = videos[j]
  497. if i != j:
  498. # 修改abCode
  499. item['abCode'] = config_.AB_CODE['w_h_rate']
  500. top2_result.append(item)
  501. new_rank_result = top2_result
  502. for i in range(len(top2_index), len(videos)):
  503. if i in flow_video_index:
  504. new_rank_result.append(flow_result[0])
  505. flow_result.pop(0)
  506. else:
  507. new_rank_result.append(other_result[0])
  508. other_result.pop(0)
  509. return new_rank_result
  510. def video_rank_with_old_video(rank_result, old_video_recall, size, top_K, old_video_index=2):
  511. """
  512. 视频分发排序 - 包含老视频, 老视频插入固定位置
  513. :param rank_result: 排序后的结果
  514. :param size: 请求数
  515. :param old_video_index: 老视频插入的位置索引,默认为2
  516. :return: new_rank_result
  517. """
  518. if not old_video_recall:
  519. return rank_result
  520. if not rank_result:
  521. return old_video_recall[:size]
  522. # 视频去重
  523. rank_video_ids = [item['videoId'] for item in rank_result]
  524. old_video_remove = []
  525. for old_video in old_video_recall:
  526. if old_video['videoId'] in rank_video_ids:
  527. old_video_remove.append(old_video)
  528. for item in old_video_remove:
  529. old_video_recall.remove(item)
  530. if not old_video_recall:
  531. return rank_result
  532. # 插入老视频
  533. # 随机获取一个视频
  534. ind = random.randint(0, len(old_video_recall) - 1)
  535. old_video = old_video_recall[ind]
  536. # 插入
  537. if len(rank_result) < top_K:
  538. new_rank_result = rank_result + [old_video]
  539. else:
  540. new_rank_result = rank_result[:old_video_index] + [old_video] + rank_result[old_video_index:]
  541. if len(new_rank_result) > size:
  542. # 判断后两位视频来源
  543. push_from_1 = new_rank_result[-1]['pushFrom']
  544. push_from_2 = new_rank_result[-2]['pushFrom']
  545. if push_from_2 == config_.PUSH_FROM['rov_recall'] and push_from_1 == config_.PUSH_FROM['flow_recall']:
  546. new_rank_result = new_rank_result[:-2] + new_rank_result[-1:]
  547. return new_rank_result[:size]
  548. def video_new_rank2(data, size, top_K, flow_pool_P, ab_code, exp_config=None):
  549. """
  550. 视频分发排序
  551. :param data: 各路召回的视频 type-dict {'rov_pool_recall': [], 'flow_pool_recall': []}
  552. :param size: 请求数
  553. :param top_K: 保证topK为召回池视频 type-int
  554. :param flow_pool_P: size-top_K视频为流量池视频的概率 type-float
  555. :return: rank_result
  556. """
  557. if not data['rov_pool_recall'] and not data['flow_pool_recall']:
  558. return [], 0
  559. #全量的是vlog,票圈精选, 334,60057,
  560. # 60054: simrecall,
  561. # 60052: 票圈精选,融合排序,60053:空置
  562. # 60055: video_reall, 60065: video_recall2
  563. # 60056: get_U2I_reall
  564. pre_str = "k_p2:"
  565. if ab_code==60057:
  566. pre_str = "k_p2:"
  567. elif ab_code == 60052:
  568. pre_str = "k_p5:"
  569. elif ab_code == 60053:
  570. pre_str = "k_p8:"
  571. elif ab_code == 60054:
  572. pre_str = "k_p3:"
  573. elif ab_code == 60055:
  574. pre_str = "k_p4:"
  575. elif ab_code == 60056:
  576. pre_str = "k_p7:"
  577. #print("pre_str:", pre_str)
  578. recall_list = []
  579. #call rank service
  580. # if ab_code == 60066:
  581. # get_featurs()
  582. #
  583. # else:
  584. redisObj = RedisHelper()
  585. vidKeys = []
  586. for recall_item in data['rov_pool_recall']:
  587. if len(recall_item) <= 0:
  588. continue
  589. vid = recall_item.get("videoId", 0)
  590. vidKeys.append(pre_str + str(vid))
  591. recall_list.append(recall_item)
  592. # print("vidKeys:", vidKeys)
  593. video_scores = redisObj.get_batch_key(vidKeys)
  594. # print("video_score:",video_scores)
  595. for i in range(len(video_scores)):
  596. try:
  597. # print(video_scores[i])
  598. if video_scores[i] is None:
  599. recall_list[i]['sort_score'] = 0.0
  600. else:
  601. video_score_str = json.loads(video_scores[i])
  602. # print("video_score_str:", video_score_str)
  603. recall_list[i]['sort_score'] = video_score_str[0]
  604. except Exception:
  605. recall_list[i]['sort_score'] = 0.0
  606. #sort_items = sorted(video_items, key=lambda k: k[1], reverse=True)
  607. rov_recall_rank =sorted(recall_list, key=lambda k: k.get('sort_score', 0), reverse=True)
  608. #print(rov_recall_rank)
  609. flow_recall_rank = sorted(data['flow_pool_recall'], key=lambda k: k.get('rovScore', 0), reverse=True)
  610. rov_recall_rank, flow_recall_rank = remove_duplicate(rov_recall=rov_recall_rank, flow_recall=flow_recall_rank,
  611. top_K=top_K)
  612. rank_result = []
  613. rank_set = set('')
  614. # 从ROV召回池中获取top k
  615. if len(rov_recall_rank) > 0:
  616. rank_result.extend(rov_recall_rank[:top_K])
  617. rov_recall_rank = rov_recall_rank[top_K:]
  618. else:
  619. rank_result.extend(flow_recall_rank[:top_K])
  620. flow_recall_rank = flow_recall_rank[top_K:]
  621. # 按概率 p 及score排序获取 size - k 个视频
  622. flow_num = 0
  623. flowConfig = 0
  624. if exp_config and exp_config['flowConfig']:
  625. flowConfig = exp_config['flowConfig']
  626. if flowConfig == 1 and len(rov_recall_rank) > 0:
  627. for recall_item in rank_result:
  628. flow_recall_name = recall_item.get("flowPool", '')
  629. flow_num = flow_num + 1
  630. all_recall_rank = rov_recall_rank + flow_recall_rank
  631. if flow_num > 0:
  632. rank_result.extend(all_recall_rank[:size - top_K])
  633. return rank_result, flow_num
  634. else:
  635. i = 0
  636. while i < size - top_K:
  637. # 随机生成[0, 1)浮点数
  638. rand = random.random()
  639. # log_.info('rand: {}'.format(rand))
  640. if rand < flow_pool_P:
  641. if flow_recall_rank:
  642. rank_result.append(flow_recall_rank[0])
  643. flow_recall_rank.remove(flow_recall_rank[0])
  644. else:
  645. rank_result.extend(rov_recall_rank[:size - top_K - i])
  646. return rank_result[:size], flow_num
  647. else:
  648. if rov_recall_rank:
  649. rank_result.append(rov_recall_rank[0])
  650. rov_recall_rank.remove(rov_recall_rank[0])
  651. else:
  652. rank_result.extend(flow_recall_rank[:size - top_K - i])
  653. return rank_result[:size], flow_num
  654. i += 1
  655. else:
  656. i = 0
  657. while i < size - top_K:
  658. # 随机生成[0, 1)浮点数
  659. rand = random.random()
  660. # log_.info('rand: {}'.format(rand))
  661. if rand < flow_pool_P:
  662. if flow_recall_rank:
  663. rank_result.append(flow_recall_rank[0])
  664. flow_recall_rank.remove(flow_recall_rank[0])
  665. else:
  666. rank_result.extend(rov_recall_rank[:size - top_K - i])
  667. return rank_result[:size], flow_num
  668. else:
  669. if rov_recall_rank:
  670. rank_result.append(rov_recall_rank[0])
  671. rov_recall_rank.remove(rov_recall_rank[0])
  672. else:
  673. rank_result.extend(flow_recall_rank[:size - top_K - i])
  674. return rank_result[:size], flow_num
  675. i += 1
  676. return rank_result[:size], flow_num
  677. def video_sanke_rank(data, size, top_K, flow_pool_P, ab_Code='', exp_config=None):
  678. """
  679. 视频分发排序
  680. :param data: 各路召回的视频 type-dict {'rov_pool_recall': [], 'flow_pool_recall': []}
  681. :param size: 请求数
  682. :param top_K: 保证topK为召回池视频 type-int
  683. :param flow_pool_P: size-top_K视频为流量池视频的概率 type-float
  684. :return: rank_result
  685. """
  686. if not data['rov_pool_recall'] and not data['flow_pool_recall'] \
  687. and len(data['u2i_recall'])==0 and len(data['w2v_recall'])==0 \
  688. and len(data['sim_recall']) == 0 and len(data['u2u2i_recall']) == 0 :
  689. return [], 0
  690. # 地域分组小时级规则更新数据
  691. recall_dict = {}
  692. region_h_recall = [item for item in data['rov_pool_recall']
  693. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_region_h']]
  694. region_h_recall_rank = sorted(region_h_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  695. recall_dict['rov_recall_region_h'] = region_h_recall_rank
  696. # 地域分组小时级更新24h规则更新数据
  697. region_24h_recall = [item for item in data['rov_pool_recall']
  698. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_region_24h']]
  699. region_24h_recall_rank = sorted(region_24h_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  700. recall_dict['rov_recall_region_24h'] = region_24h_recall_rank
  701. # 相对24h规则更新数据
  702. rule_24h_recall = [item for item in data['rov_pool_recall']
  703. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_24h']]
  704. rule_24h_recall_rank = sorted(rule_24h_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  705. recall_dict['rov_recall_24h'] = rule_24h_recall_rank
  706. # 相对24h规则筛选后剩余更新数据
  707. rule_24h_dup_recall = [item for item in data['rov_pool_recall']
  708. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_24h_dup']]
  709. rule_24h_dup_recall_rank = sorted(rule_24h_dup_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  710. recall_dict['rov_recall_24h_dup'] = rule_24h_dup_recall_rank
  711. hot_recall = []
  712. w2v_recall =[]
  713. sim_recall = []
  714. u2u2i_recall = []
  715. if ab_Code==60058:
  716. if len(data['u2i_recall'])>0:
  717. hot_recall = sorted(data['u2i_recall'], key=lambda k: k.get('rovScore', 0), reverse=True)
  718. recall_dict['u2i_recall'] = hot_recall
  719. elif ab_Code==60059:
  720. if len(data['w2v_recall'])>0:
  721. recall_dict['w2v_recall'] = data['w2v_recall']
  722. else:
  723. recall_dict['w2v_recall'] = w2v_recall
  724. elif ab_Code==60061 or ab_Code==60063:
  725. if len(data['sim_recall'])>0:
  726. recall_dict['sim_recall'] = data['sim_recall']
  727. else:
  728. recall_dict['sim_recall'] = sim_recall
  729. elif ab_Code==60062:
  730. if len(data['u2u2i_recall'])>0:
  731. recall_dict['u2u2i_recall'] = data['u2u2i_recall']
  732. else:
  733. recall_dict['u2u2i_recall'] = u2u2i_recall
  734. recall_list = [('rov_recall_region_h',1, 1),('rov_recall_region_h',0.5, 1),('rov_recall_region_24h',1,1),
  735. ('u2i_recall',0.5,1), ('w2v_recall',0.5,1),('rov_recall_24h',1,1), ('rov_recall_24h_dup',0.5,1)]
  736. if exp_config and exp_config['recall_list']:
  737. recall_list = exp_config['recall_list']
  738. #print("recall_config:", recall_list)
  739. rov_recall_rank = []
  740. select_ids = set('')
  741. for i in range(3):
  742. if len(rov_recall_rank)>8:
  743. break
  744. for per_recall_item in recall_list:
  745. per_recall_name = per_recall_item[0]
  746. per_recall_freq = per_recall_item[1]
  747. per_limt_num = per_recall_item[2]
  748. rand_num = random.random()
  749. #print(recall_dict[per_recall_name])
  750. if rand_num<per_recall_freq and per_recall_name in recall_dict:
  751. per_recall = recall_dict[per_recall_name]
  752. #print("per_recall_item:", per_recall_item)
  753. cur_recall_num = 0
  754. for recall_item in per_recall:
  755. vid = recall_item['videoId']
  756. if vid in select_ids:
  757. continue
  758. rov_recall_rank.append(recall_item)
  759. select_ids.add(vid)
  760. cur_recall_num+=1
  761. if cur_recall_num>=per_limt_num:
  762. break
  763. # print("rov_recall_rank:")
  764. # print(rov_recall_rank)
  765. #rov_recall_rank = region_h_recall_rank + region_24h_recall_rank + \
  766. # rule_24h_recall_rank + rule_24h_dup_recall_rank
  767. # 流量池
  768. flow_recall_rank = sorted(data['flow_pool_recall'], key=lambda k: k.get('rovScore', 0), reverse=True)
  769. # 对各路召回的视频进行去重
  770. rov_recall_rank, flow_recall_rank = remove_duplicate(rov_recall=rov_recall_rank, flow_recall=flow_recall_rank,
  771. top_K=top_K)
  772. # log_.info('remove_duplicate finished! rov_recall_rank = {}, flow_recall_rank = {}'.format(
  773. # rov_recall_rank, flow_recall_rank))
  774. # rank_result = relevant_recall_rank
  775. rank_result = []
  776. # 从ROV召回池中获取top k
  777. if len(rov_recall_rank) > 0:
  778. rank_result.extend(rov_recall_rank[:top_K])
  779. rov_recall_rank = rov_recall_rank[top_K:]
  780. else:
  781. rank_result.extend(flow_recall_rank[:top_K])
  782. flow_recall_rank = flow_recall_rank[top_K:]
  783. flow_num = 0
  784. flowConfig =0
  785. if exp_config and exp_config['flowConfig']:
  786. flowConfig = exp_config['flowConfig']
  787. if flowConfig == 1 and len(rov_recall_rank) > 0:
  788. rank_result.extend(rov_recall_rank[:top_K])
  789. for recall_item in rank_result:
  790. flow_recall_name = recall_item.get("flowPool", '')
  791. if flow_recall_name is not None and flow_recall_name.find("#") > -1:
  792. flow_num = flow_num + 1
  793. all_recall_rank = rov_recall_rank + flow_recall_rank
  794. if flow_num > 0:
  795. rank_result.extend(all_recall_rank[:size - top_K])
  796. return rank_result[:size], flow_num
  797. else:
  798. # 按概率 p 及score排序获取 size - k 个视频
  799. i = 0
  800. while i < size - top_K:
  801. # 随机生成[0, 1)浮点数
  802. rand = random.random()
  803. # log_.info('rand: {}'.format(rand))
  804. if rand < flow_pool_P:
  805. if flow_recall_rank:
  806. rank_result.append(flow_recall_rank[0])
  807. flow_recall_rank.remove(flow_recall_rank[0])
  808. else:
  809. rank_result.extend(rov_recall_rank[:size - top_K - i])
  810. return rank_result[:size], flow_num
  811. else:
  812. if rov_recall_rank:
  813. rank_result.append(rov_recall_rank[0])
  814. rov_recall_rank.remove(rov_recall_rank[0])
  815. else:
  816. rank_result.extend(flow_recall_rank[:size - top_K - i])
  817. return rank_result[:size], flow_num
  818. i += 1
  819. else:
  820. # 按概率 p 及score排序获取 size - k 个视频
  821. i = 0
  822. while i < size - top_K:
  823. # 随机生成[0, 1)浮点数
  824. rand = random.random()
  825. # log_.info('rand: {}'.format(rand))
  826. if rand < flow_pool_P:
  827. if flow_recall_rank:
  828. rank_result.append(flow_recall_rank[0])
  829. flow_recall_rank.remove(flow_recall_rank[0])
  830. else:
  831. rank_result.extend(rov_recall_rank[:size - top_K - i])
  832. return rank_result[:size], flow_num
  833. else:
  834. if rov_recall_rank:
  835. rank_result.append(rov_recall_rank[0])
  836. rov_recall_rank.remove(rov_recall_rank[0])
  837. else:
  838. rank_result.extend(flow_recall_rank[:size - top_K - i])
  839. return rank_result[:size],flow_num
  840. i += 1
  841. return rank_result[:size], flow_num
  842. def video_sank_pos_rank(data, size, top_K, flow_pool_P, ab_Code='', exp_config=None):
  843. """
  844. 视频分发排序
  845. :param data: 各路召回的视频 type-dict {'rov_pool_recall': [], 'flow_pool_recall': []}
  846. :param size: 请求数
  847. :param top_K: 保证topK为召回池视频 type-int
  848. :param flow_pool_P: size-top_K视频为流量池视频的概率 type-float
  849. :return: rank_result
  850. """
  851. if not data['rov_pool_recall'] and not data['flow_pool_recall'] \
  852. and len(data['u2i_recall'])==0 and len(data['w2v_recall'])==0 \
  853. and len(data['sim_recall']) == 0 and len(data['u2u2i_recall']) == 0 :
  854. return [], 0
  855. # 地域分组小时级规则更新数据
  856. recall_dict = {}
  857. region_h_recall = [item for item in data['rov_pool_recall']
  858. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_region_h']]
  859. region_h_recall_rank = sorted(region_h_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  860. recall_dict['rov_recall_region_h'] = region_h_recall_rank
  861. # 地域分组小时级更新24h规则更新数据
  862. region_24h_recall = [item for item in data['rov_pool_recall']
  863. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_region_24h']]
  864. region_24h_recall_rank = sorted(region_24h_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  865. recall_dict['rov_recall_region_24h'] = region_24h_recall_rank
  866. # 相对24h规则更新数据
  867. rule_24h_recall = [item for item in data['rov_pool_recall']
  868. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_24h']]
  869. rule_24h_recall_rank = sorted(rule_24h_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  870. recall_dict['rov_recall_24h'] = rule_24h_recall_rank
  871. # 相对24h规则筛选后剩余更新数据
  872. rule_24h_dup_recall = [item for item in data['rov_pool_recall']
  873. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_24h_dup']]
  874. rule_24h_dup_recall_rank = sorted(rule_24h_dup_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  875. recall_dict['rov_recall_24h_dup'] = rule_24h_dup_recall_rank
  876. u2i_recall = []
  877. u2i_play_recall = []
  878. w2v_recall =[]
  879. sim_recall = []
  880. u2u2i_recall = []
  881. return_video_recall = []
  882. #print("")
  883. if ab_Code==60058:
  884. if len(data['u2i_recall'])>0:
  885. recall_dict['u2i_recall'] = data['u2i_recall']
  886. else:
  887. recall_dict['u2i_recall'] = u2i_recall
  888. if len(data['u2i_play_recall']) > 0:
  889. recall_dict['u2i_play_recall'] = data['u2i_play_recall']
  890. else:
  891. recall_dict['u2i_play_recall'] = u2i_play_recall
  892. elif ab_Code==60059:
  893. if len(data['w2v_recall'])>0:
  894. recall_dict['w2v_recall'] = data['w2v_recall']
  895. else:
  896. recall_dict['w2v_recall'] = w2v_recall
  897. elif ab_Code==60061 or ab_Code==60063:
  898. if len(data['sim_recall'])>0:
  899. recall_dict['sim_recall'] = data['sim_recall']
  900. else:
  901. recall_dict['sim_recall'] = sim_recall
  902. elif ab_Code==60062:
  903. if len(data['u2u2i_recall'])>0:
  904. recall_dict['u2u2i_recall'] = data['u2u2i_recall']
  905. else:
  906. recall_dict['u2u2i_recall'] = u2u2i_recall
  907. elif ab_Code==60064:
  908. if len(data['return_video_recall'])>0:
  909. recall_dict['return_video_recall'] = data['return_video_recall']
  910. else:
  911. recall_dict['return_video_recall'] = return_video_recall
  912. recall_pos1 = [('rov_recall_region_h',0, 0.98),('rov_recall_24h',0.98, 1),('rov_recall_region_24h',0,1),
  913. ('rov_recall_24h',0,1), ('rov_recall_24h_dup',0,1)]
  914. recall_pos2 = [('rov_recall_region_h',0,0.98),('rov_recall_24h',0.98,1),('rov_recall_region_24h',0,1),
  915. ('rov_recall_24h',0,1),('rov_recall_24h_dup',0,1)]
  916. recall_pos3 = [('rov_recall_region_h', 0,0.98), ('rov_recall_24h', 0.98,1), ('rov_recall_region_24h', 0,1),
  917. ('rov_recall_24h', 0,1), ('rov_recall_24h_dup', 0,1)]
  918. recall_pos4 = [('rov_recall_region_h', 0,0.98), ('rov_recall_24h', 0,0.02), ('rov_recall_region_24h', 0,1),
  919. ('rov_recall_24h', 0,1), ('rov_recall_24h_dup', 0,1)]
  920. if exp_config and 'recall_pos1' in exp_config \
  921. and 'recall_pos2' in exp_config \
  922. and 'recall_pos3' in exp_config \
  923. and 'recall_pos4' in exp_config :
  924. recall_pos1 = exp_config['recall_pos1']
  925. recall_pos2 = exp_config['recall_pos2']
  926. recall_pos3 = exp_config['recall_pos3']
  927. recall_pos4 = exp_config['recall_pos4']
  928. #print("recall_config:", recall_pos1)
  929. rov_recall_rank = []
  930. recall_list = []
  931. recall_list.append(recall_pos1)
  932. recall_list.append(recall_pos2)
  933. recall_list.append(recall_pos3)
  934. recall_list.append(recall_pos4)
  935. select_ids = set('')
  936. recall_num_limit_dict = {}
  937. if exp_config and 'recall_num_limit' in exp_config:
  938. recall_num_limit_dict = exp_config['recall_num_limit']
  939. exp_recall_dict = {}
  940. #index_pos = 0
  941. for j in range(3):
  942. if len(rov_recall_rank)>12:
  943. break
  944. # choose pos
  945. for recall_pos_config in recall_list:
  946. rand_num = random.random()
  947. index_pos = 0
  948. # choose pos recall
  949. for per_recall_item in recall_pos_config:
  950. if index_pos == 1:
  951. break
  952. if len(per_recall_item)<3:
  953. continue
  954. per_recall_name = per_recall_item[0]
  955. per_recall_min = float(per_recall_item[1])
  956. per_recall_max = float(per_recall_item[2])
  957. per_recall_num = exp_recall_dict.get(per_recall_name, 0)
  958. per_recall_total_num = recall_num_limit_dict.get(per_recall_name, 0)
  959. # recall set total num
  960. if len(recall_num_limit_dict)>0 and per_recall_total_num>0 and per_recall_num>= per_recall_total_num:
  961. continue
  962. if rand_num >= per_recall_min and rand_num < per_recall_max and per_recall_name in recall_dict:
  963. per_recall = recall_dict[per_recall_name]
  964. for recall_item in per_recall:
  965. vid = recall_item['videoId']
  966. if vid in select_ids:
  967. continue
  968. recall_item['rand'] = rand_num
  969. rov_recall_rank.append(recall_item)
  970. select_ids.add(vid)
  971. if per_recall_name in exp_recall_dict:
  972. exp_recall_dict[per_recall_name] +=1
  973. else:
  974. exp_recall_dict[per_recall_name] = 1
  975. index_pos = 1
  976. break
  977. #print("rov_recall_rank:", rov_recall_rank)
  978. if len(rov_recall_rank)<4:
  979. rov_doudi_rank = region_h_recall_rank + sim_recall + u2i_recall + u2u2i_recall + w2v_recall +return_video_recall+u2i_play_recall+ region_24h_recall_rank + rule_24h_recall_rank + rule_24h_dup_recall_rank
  980. for recall_item in rov_doudi_rank:
  981. vid = recall_item['videoId']
  982. if vid in select_ids:
  983. continue
  984. rov_recall_rank.append(recall_item)
  985. select_ids.add(vid)
  986. if len(rov_recall_rank)>12:
  987. break
  988. # print("rov_recall_rank:")
  989. #print(rov_recall_rank)
  990. # 流量池
  991. flow_recall_rank = sorted(data['flow_pool_recall'], key=lambda k: k.get('rovScore', 0), reverse=True)
  992. # 对各路召回的视频进行去重
  993. rov_recall_rank, flow_recall_rank = remove_duplicate(rov_recall=rov_recall_rank, flow_recall=flow_recall_rank,
  994. top_K=top_K)
  995. # log_.info('remove_duplicate finished! rov_recall_rank = {}, flow_recall_rank = {}'.format(
  996. # rov_recall_rank, flow_recall_rank))
  997. # rank_result = relevant_recall_rank
  998. rank_result = []
  999. # 从ROV召回池中获取top k
  1000. if len(rov_recall_rank) > 0:
  1001. rank_result.extend(rov_recall_rank[:top_K])
  1002. rov_recall_rank = rov_recall_rank[top_K:]
  1003. else:
  1004. rank_result.extend(flow_recall_rank[:top_K])
  1005. flow_recall_rank = flow_recall_rank[top_K:]
  1006. flow_num = 0
  1007. flowConfig =0
  1008. if exp_config and exp_config['flowConfig']:
  1009. flowConfig = exp_config['flowConfig']
  1010. if flowConfig == 1 and len(rov_recall_rank) > 0:
  1011. rank_result.extend(rov_recall_rank[:top_K])
  1012. for recall_item in rank_result:
  1013. flow_recall_name = recall_item.get("flowPool", '')
  1014. if flow_recall_name is not None and flow_recall_name.find("#") > -1:
  1015. flow_num = flow_num + 1
  1016. all_recall_rank = rov_recall_rank + flow_recall_rank
  1017. if flow_num > 0:
  1018. rank_result.extend(all_recall_rank[:size - top_K])
  1019. return rank_result[:size], flow_num
  1020. else:
  1021. # 按概率 p 及score排序获取 size - k 个视频
  1022. i = 0
  1023. while i < size - top_K:
  1024. # 随机生成[0, 1)浮点数
  1025. rand = random.random()
  1026. # log_.info('rand: {}'.format(rand))
  1027. if rand < flow_pool_P:
  1028. if flow_recall_rank:
  1029. rank_result.append(flow_recall_rank[0])
  1030. flow_recall_rank.remove(flow_recall_rank[0])
  1031. else:
  1032. rank_result.extend(rov_recall_rank[:size - top_K - i])
  1033. return rank_result[:size], flow_num
  1034. else:
  1035. if rov_recall_rank:
  1036. rank_result.append(rov_recall_rank[0])
  1037. rov_recall_rank.remove(rov_recall_rank[0])
  1038. else:
  1039. rank_result.extend(flow_recall_rank[:size - top_K - i])
  1040. return rank_result[:size], flow_num
  1041. i += 1
  1042. else:
  1043. # 按概率 p 及score排序获取 size - k 个视频
  1044. i = 0
  1045. while i < size - top_K:
  1046. # 随机生成[0, 1)浮点数
  1047. rand = random.random()
  1048. # log_.info('rand: {}'.format(rand))
  1049. if rand < flow_pool_P:
  1050. if flow_recall_rank:
  1051. rank_result.append(flow_recall_rank[0])
  1052. flow_recall_rank.remove(flow_recall_rank[0])
  1053. else:
  1054. rank_result.extend(rov_recall_rank[:size - top_K - i])
  1055. return rank_result[:size], flow_num
  1056. else:
  1057. if rov_recall_rank:
  1058. rank_result.append(rov_recall_rank[0])
  1059. rov_recall_rank.remove(rov_recall_rank[0])
  1060. else:
  1061. rank_result.extend(flow_recall_rank[:size - top_K - i])
  1062. return rank_result[:size],flow_num
  1063. i += 1
  1064. return rank_result[:size], flow_num
  1065. if __name__ == '__main__':
  1066. d_test = [{'videoId': 10028734, 'rovScore': 99.977, 'pushFrom': 'recall_pool', 'abCode': 10000},
  1067. {'videoId': 1919925, 'rovScore': 99.974, 'pushFrom': 'recall_pool', 'abCode': 10000},
  1068. {'videoId': 9968118, 'rovScore': 99.972, 'pushFrom': 'recall_pool', 'abCode': 10000},
  1069. {'videoId': 9934863, 'rovScore': 99.971, 'pushFrom': 'recall_pool', 'abCode': 10000},
  1070. {'videoId': 10219869, 'flowPool': '1#1#1#1640830818883', 'rovScore': 82.21929728934731, 'pushFrom': 'flow_pool', 'abCode': 10000},
  1071. {'videoId': 10212814, 'flowPool': '1#1#1#1640759014984', 'rovScore': 81.26694187726412, 'pushFrom': 'flow_pool', 'abCode': 10000},
  1072. {'videoId': 10219437, 'flowPool': '1#1#1#1640827620520', 'rovScore': 81.21634156641908, 'pushFrom': 'flow_pool', 'abCode': 10000},
  1073. {'videoId': 1994050, 'rovScore': 99.97, 'pushFrom': 'recall_pool', 'abCode': 10000},
  1074. {'videoId': 9894474, 'rovScore': 99.969, 'pushFrom': 'recall_pool', 'abCode': 10000},
  1075. {'videoId': 10028081, 'rovScore': 99.966, 'pushFrom': 'recall_pool', 'abCode': 10000}]
  1076. res = video_rank_by_w_h_rate(videos=d_test)
  1077. for tmp in res:
  1078. print(tmp)