liqian hace 2 años
padre
commit
55001f6b08
Se han modificado 2 ficheros con 5 adiciones y 3 borrados
  1. 2 1
      ad_recommend.py
  2. 3 2
      app.py

+ 2 - 1
ad_recommend.py

@@ -49,4 +49,5 @@ def ad_recommend_predict(mid, video_id):
         ad_predict = 1
     else:
         ad_predict = 0
-    return ad_predict
+    result = {'mid_video_share_rate': mid_video_share_rate, 'threshold': threshold, 'ad_predict': ad_predict}
+    return result

+ 3 - 2
app.py

@@ -298,13 +298,14 @@ def ad_predict():
         request_data = json.loads(request.get_data())
         mid = request_data.get('mid')
         video_id = request_data.get('videoId')
-        ad_predict = ad_recommend_predict(mid=mid, video_id=video_id)
-        result = {'code': 200, 'message': 'success', 'data': ad_predict}
+        predict_result = ad_recommend_predict(mid=mid, video_id=video_id)
+        result = {'code': 200, 'message': 'success', 'data': predict_result.get('ad_predict')}
         log_message = {
             'requestUri': '/applet/ad/predict',
             'logTimestamp': int(time.time() * 1000),
             'mid': mid,
             'video_id': video_id,
+            'predict_result': predict_result,
             'result': result,
             'executeTime': (time.time() - start_time) * 1000
         }