Fix regression test failure: TestSearchIntegration.test_cross_validate (

#958) * fixed the bug on cross_validation step from getting max value by column to getting max value by row
castorini · Jan 18, 2022 · c38c557 · c38c557
1 parent 2df246f
commit c38c557
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/classifier_prf/cross_validate.py b/scripts/classifier_prf/cross_validate.py
@@ -92,7 +92,7 @@ def generate_run_file(folders, df, collection, run_file, classifier, rm3, output
                               for f in folders for topic in f if f != folder and str(topic) in df.index]
             train_df = df.loc[train_topicids, :]
             train_df.loc['Mean', :] = train_df.mean(axis=0)
-            highest_alpha = train_df.iloc[-1, :].idxmax(axis=1)
+            highest_alpha = train_df.iloc[-1, :].idxmax(axis=0)
             highest_alpha_lst.append(highest_alpha)
 
             for topic in folder: