added extra tests

madgik · Jan 12, 2025 · ee1805f · ee1805f
1 parent 5ba597d
commit ee1805f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/exareme2/algorithms/flower/xgboost/server.py b/exareme2/algorithms/flower/xgboost/server.py
@@ -34,10 +34,10 @@ def aggregate_evaluate(self, rnd, results, failures):
             self.initial_auc = aggregated_metrics["AUC"]
         if rnd == self.num_rounds:
             print(aggregated_metrics)
-            curr_auc = aggregated_metrics["AUC"]
+            curr_auc = aggregated_metrics[1]["AUC"]
             auc_diff = curr_auc - self.initial_auc
             auc_ascending = ""
-            if auc_diff > 0.0:
+            if auc_diff >= 0.0:
                 auc_ascending = "correct"
             else:
                 auc_ascending = "not_correct"