We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a310575 commit 65c8727Copy full SHA for 65c8727
1 file changed
_unittests/ut_ml/test_neural_tree.py
@@ -417,12 +417,12 @@ def test_neural_net_gradient_regression_2_h2(self):
417
for p in range(0, 5):
418
grad1 = neu.gradient(X[p], y[p])
419
grad2 = net.gradient(X[p], y[p])
420
- self.assertEqualArray(grad1, grad2[:3])
+ self.assertEqualArray(grad1, grad2[:3], atol=1e-7)
421
422
loss1 = net.loss(X, y).sum()
423
net.fit(X, y, max_iter=20)
424
loss2 = net.loss(X, y).sum()
425
- self.assertLess(loss2, loss1, atol=1e-7)
+ self.assertLess(loss2, loss1 + 1e-7)
426
427
def test_neural_net_gradient_fit(self):
428
X = numpy.arange(16).astype(numpy.float64).reshape((-1, 2))
0 commit comments