Skip to content

Commit 8270581

Browse files
committed
Update test_neural_tree.py
1 parent 4f12398 commit 8270581

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
@@ -355,12 +355,12 @@ def test_neural_net_gradient_regression_2(self):
355355
self.assertEqualArray(pred1, pred2[:, -1])
356356
self.assertEqualArray(pred2[:, -2], pred2[:, -1])
357357
self.assertEqualArray(pred2[:, 2], pred2[:, 3])
358-
self.assertEqualArray(loss1, loss2)
358+
self.assertEqualArray(loss1, loss2, atol=1e-7)
359359

360360
for p in range(0, 5):
361361
grad1 = neu.gradient(X[p], y[p])
362362
grad2 = net.gradient(X[p], y[p])
363-
self.assertEqualArray(grad1, grad2[:3])
363+
self.assertEqualArray(grad1, grad2[:3], atol=1e-7)
364364

365365
def test_neural_net_gradient_regression_2_h2(self):
366366
X = numpy.abs(numpy.random.randn(10, 2))

0 commit comments

Comments
 (0)