|
@@ -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)
|
|
|
|
|
|
|