liqian 3 年之前
父节点
当前提交
9d322ca88d
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 2 1
      rov_train.py

+ 2 - 1
rov_train.py

@@ -181,7 +181,7 @@ def predict_test():
     video_ids = [video[0] for video in data]
     # 视频状态过滤
     filtered_videos = filter_video_status(video_ids)
-    log_.info('filtered_videos nums={}'.format(len(filtered_videos)))
+    log_.info('filtered_videos count = {}'.format(len(filtered_videos)))
     # 随机生成 0-100 数作为分数
     redis_data = {}
     json_data = []
@@ -189,6 +189,7 @@ def predict_test():
         score = random.uniform(0, 100)
         redis_data[video_id] = score
         json_data.append({'videoId': video_id, 'rovScore': score})
+    log_.info('json_data count = {}'.format(len(json_data)))
     # 上传Redis
     redis_helper = RedisHelper()
     key_name = config_.RECALL_KEY_NAME_PREFIX + time.strftime('%Y%m%d')