Update launch_xgboost.py

mmcdermott · Jul 17, 2024 · 48550d1 · 48550d1
1 parent 4039a96
commit 48550d1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/MEDS_tabular_automl/scripts/launch_xgboost.py b/src/MEDS_tabular_automl/scripts/launch_xgboost.py
@@ -399,13 +399,13 @@ def _build_iterators(self):
 
     @TimeableMixin.TimeAs
     def evaluate(self) -> float:
-        """Evaluates the model on the test set.
+        """Evaluates the model on the tuning set.
 
         Returns:
             The evaluation metric as the ROC AUC score.
         """
-        y_pred = self.model.predict(self.dheld_out)
-        y_true = self.dheld_out.get_label()
+        y_pred = self.model.predict(self.dtuning)
+        y_true = self.dtuning.get_label()
         return roc_auc_score(y_true, y_pred)