Browse Source

feat:添加评估结果分析脚本

zhaohaipeng 6 months ago
parent
commit
101aebe9c7
1 changed files with 2 additions and 2 deletions
  1. 2 2
      ad/model_predict_analyse.py

+ 2 - 2
ad/model_predict_analyse.py

@@ -38,13 +38,13 @@ def _main(model1_predict_path: str, model2_predict_path: str):
     m1 = pd.DataFrame(model1_result)
     g1 = m1.groupby("cid").agg(count=('cid', 'size'), average_value=('score', 'mean'))
     # 获取出现次数最多的十个 cid
-    most_common_cid1 = g1.nlargest(10, 'count')
+    most_common_cid1 = g1.nlargest(1000, 'count')
     print(most_common_cid1)
 
     m2 = pd.DataFrame(model2_result)
     g2 = m2.groupby("cid").agg(count=('cid', 'size'), average_value=('score', 'mean'))
     # 获取出现次数最多的十个 cid
-    most_common_cid2 = g2.nlargest(10, 'count')
+    most_common_cid2 = g2.nlargest(1000, 'count')
     print(most_common_cid2)