Merge pull request #1043 from Trusted-AI/dependabot/pip/xgboost-1.4.0

Bump xgboost from 1.3.3 to 1.4.0

Author: beat-buesser

requirements.txt

@@ -31,7 +31,7 @@ torchvision==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
 catboost==0.25.1
 GPy==1.9.9
 lightgbm==3.2.1
-xgboost==1.3.3
+xgboost==1.4.0
 kornia~=0.5.0
 
 # Lingvo ASR dependencies

tests/metrics/test_verification_decision_trees.py

@@ -61,7 +61,7 @@ def setUp(cls):
         master_seed(seed=42)
 
     def test_XGBoost(self):
-        model = XGBClassifier(n_estimators=4, max_depth=6)
+        model = XGBClassifier(n_estimators=4, max_depth=6, objective="multi:softprob",  eval_metric="merror")
         model.fit(self.x_train, np.argmax(self.y_train, axis=1))
 
         classifier = XGBoostClassifier(model=model, nb_features=self.n_features, nb_classes=self.n_classes)
@@ -71,8 +71,8 @@ def test_XGBoost(self):
             x=self.x_test, y=self.y_test, eps_init=0.3, nb_search_steps=10, max_clique=2, max_level=2
         )
 
-        self.assertEqual(average_bound, 0.03186914062500001)
-        self.assertEqual(verified_error, 0.99)
+        self.assertEqual(average_bound, 0.0011425781249999997)
+        self.assertEqual(verified_error, 1.0)
 
     def test_LightGBM(self):
         train_data = lightgbm.Dataset(self.x_train, label=np.argmax(self.y_train, axis=1))