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. 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, mid, exp_config=None, env_dict=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. rov_recall_rank = data['rov_pool_recall']
  580. #call rank service
  581. if ab_code == 60066:
  582. feature_dict, recall_list = get_featurs(mid, data, size, top_K, flow_pool_P, env_dict)
  583. score_result = get_tf_serving_sores(feature_dict)
  584. if score_result and len(score_result) > 0 and len(score_result) == len(recall_list):
  585. for i in range(len(score_result)):
  586. recall_list[i]['sort_score'] = score_result[i][0]
  587. #print(score_result[i][0])
  588. rov_recall_rank = sorted(recall_list, key=lambda k: k.get('sort_score', 0), reverse=True)
  589. else:
  590. redisObj = RedisHelper()
  591. vidKeys = []
  592. if video_scores and len(recall_list) > 0:
  593. for i in range(len(video_scores)):
  594. try:
  595. if video_scores[i] is None:
  596. recall_list[i]['sort_score'] = 0.0
  597. else:
  598. video_score_str = json.loads(video_scores[i])
  599. recall_list[i]['sort_score'] = video_score_str[0]
  600. except Exception:
  601. recall_list[i]['sort_score'] = 0.0
  602. rov_recall_rank = sorted(recall_list, key=lambda k: k.get('sort_score', 0), reverse=True)
  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)