Update test related to LightGBM

Beat Buesser · Beat Buesser · commit 55cb3378dbab · 2020-09-04T15:57:55.000+01:00
Signed-off-by: Beat Buesser &lt;beat.buesser@ie.ibm.com&gt;
diff --git a/tests/estimators/classification/test_lightgbm.py b/tests/estimators/classification/test_lightgbm.py
@@ -48,7 +48,7 @@ def setUpClass(cls):
 
     def test_predict(self):
         y_predicted = self.classifier.predict(self.x_test_iris[0:1])
-        y_expected = np.asarray([[0.14644158, 0.16454982, 0.68900861]])
+        y_expected = np.asarray([[0.1083, 0.1255, 0.7663]])
         np.testing.assert_array_almost_equal(y_predicted, y_expected, decimal=4)
 
 
diff --git a/tests/metrics/test_verification_decision_trees.py b/tests/metrics/test_verification_decision_trees.py
@@ -103,8 +103,8 @@ def test_LightGBM(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.07634765624999999)
-        self.assertEqual(verified_error, 0.85)
+        self.assertEqual(average_bound, 0.047742187500000005)
+        self.assertEqual(verified_error, 0.94)
 
     def test_GradientBoosting(self):
         model = GradientBoostingClassifier(n_estimators=4, max_depth=6)

Original file line number	Diff line number	Diff line change
`@@ -103,8 +103,8 @@ def test_LightGBM(self):`
`103`	`103`	`x=self.x_test, y=self.y_test, eps_init=0.3, nb_search_steps=10, max_clique=2, max_level=2`
`104`	`104`	`)`
`105`	`105`
`106`		`- self.assertEqual(average_bound, 0.07634765624999999)`
`107`		`- self.assertEqual(verified_error, 0.85)`
	`106`	`+ self.assertEqual(average_bound, 0.047742187500000005)`
	`107`	`+ self.assertEqual(verified_error, 0.94)`
`108`	`108`
`109`	`109`	`def test_GradientBoosting(self):`
`110`	`110`	`model = GradientBoostingClassifier(n_estimators=4, max_depth=6)`