Updated xgboost with eval_set

SeldonIO · Jul 4, 2024 · d350dac · d350dac
1 parent 73a921e
commit d350dac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/runtimes/xgboost/tests/conftest.py b/runtimes/xgboost/tests/conftest.py
@@ -50,7 +50,7 @@ def classifier_uri(tmp_path) -> str:
     clf = xgb.XGBClassifier(
         num_class=c, use_label_encoder=False, objective="multi:softprob"
     )
-    clf.fit(X_train, y_train, eval_metric="mlogloss")
+    clf.fit(X_train, y_train, eval_set=[(X_train, y_train)])
 
     model_uri = os.path.join(tmp_path, "xgboost-model.json")
     clf.save_model(model_uri)