Skip to content

Commit 65c8727

Browse files
committed
Update test_neural_tree.py
1 parent a310575 commit 65c8727

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

_unittests/ut_ml/test_neural_tree.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -417,12 +417,12 @@ def test_neural_net_gradient_regression_2_h2(self):
417417
for p in range(0, 5):
418418
grad1 = neu.gradient(X[p], y[p])
419419
grad2 = net.gradient(X[p], y[p])
420-
self.assertEqualArray(grad1, grad2[:3])
420+
self.assertEqualArray(grad1, grad2[:3], atol=1e-7)
421421

422422
loss1 = net.loss(X, y).sum()
423423
net.fit(X, y, max_iter=20)
424424
loss2 = net.loss(X, y).sum()
425-
self.assertLess(loss2, loss1, atol=1e-7)
425+
self.assertLess(loss2, loss1 + 1e-7)
426426

427427
def test_neural_net_gradient_fit(self):
428428
X = numpy.arange(16).astype(numpy.float64).reshape((-1, 2))

0 commit comments

Comments
 (0)