liqian 1 year ago
parent
commit
d6d1bc76e4
1 changed files with 5 additions and 1 deletions
  1. 5 1
      video_recall.py

+ 5 - 1
video_recall.py

@@ -1061,8 +1061,12 @@ class PoolRecall(object):
                 weight_sum = sum([int(item[2]) for item in available_level])
                 level_p_mapping = {}
                 level_p_low = 0
-                for level, level_key, weight in available_level:
+                available_level_count = len(available_level)
+                for i, item in enumerate(available_level):
+                    level, level_key, weight = item[0], item[1], item[2]
                     level_p_up = level_p_low + weight/weight_sum
+                    if i == available_level_count:
+                        level_p_up = 1
                     level_p_mapping[level] = {
                         'key': level_key,
                         'level_p': [round(level_p_low, 2), round(level_p_up, 2)]