video_rank.py 50 KB

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