Skip to content

Commit

Permalink
Update workflows for TensorFlow/Keras 2.14
Browse files Browse the repository at this point in the history
Signed-off-by: Beat Buesser <[email protected]>
  • Loading branch information
beat-buesser committed Nov 16, 2023
1 parent abf4727 commit 4f302fe
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/estimators/classification/test_scikitlearn.py
Original file line number Diff line number Diff line change
Expand Up @@ -315,7 +315,7 @@ def test_class_gradient_int_2(self):

def test_class_gradient_list_1(self):
grad_predicted = self.classifier.class_gradient(self.x_test_iris[0:1], label=[1])
grad_expected = [[[-0.56317311, -0.70493763, -0.98874801, -0.67053026]]]
grad_expected = [[[-0.56317311, -0.70493763, -0.98874801, -0.67106276]]]

for i_shape in range(4):
self.assertAlmostEqual(grad_predicted[0, 0, i_shape], grad_expected[0][0][i_shape], 3)
Expand Down

0 comments on commit 4f302fe

Please sign in to comment.