Explorar o código

update flowpool video score

liqian %!s(int64=3) %!d(string=hai) anos
pai
achega
bc32988452
Modificáronse 1 ficheiros con 3 adicións e 1 borrados
  1. 3 1
      pool_predict.py

+ 3 - 1
pool_predict.py

@@ -1,6 +1,7 @@
 import time
 import os
 import traceback
+import random
 
 from config import set_config
 from utils import request_post, filter_video_status, send_msg_to_feishu
@@ -61,7 +62,8 @@ def get_videos_remain_view_count(app_type, videos_info):
 
 
 def get_score(video_ids):
-    return [1] * len(video_ids)
+    # 以[0, 100]之间的随机浮点数作为score
+    return [random.uniform(0, 100) for _ in range(len(video_ids))]
 
 
 def predict(app_type):