liqian hace 2 años
padre
commit
ad9df10444
Se han modificado 1 ficheros con 7 adiciones y 1 borrados
  1. 7 1
      ad_recommend.py

+ 7 - 1
ad_recommend.py

@@ -49,5 +49,11 @@ def ad_recommend_predict(mid, video_id):
         ad_predict = 2
     else:
         ad_predict = 1
-    result = {'mid_video_share_rate': mid_video_share_rate, 'threshold': threshold, 'ad_predict': ad_predict}
+    result = {
+        'mid_group': mid_group,
+        'group_share_rate': group_share_rate,
+        'video_share_rate': video_share_rate,
+        'mid_video_share_rate': mid_video_share_rate,
+        'threshold': threshold,
+        'ad_predict': ad_predict}
     return result