diff --git a/h2o-algos/src/test/java/hex/deeplearning/DeepLearningSHAPTest.java b/h2o-algos/src/test/java/hex/deeplearning/DeepLearningSHAPTest.java index 8249d7912c11..bd9b804bc756 100644 --- a/h2o-algos/src/test/java/hex/deeplearning/DeepLearningSHAPTest.java +++ b/h2o-algos/src/test/java/hex/deeplearning/DeepLearningSHAPTest.java @@ -66,7 +66,7 @@ public void testClassificationCompactSHAP() { Val val = Rapids.exec("(sumaxis " + contribs._key + " 0 1)"); assertTrue(val instanceof ValFrame); res = val.getFrame(); - assertColsEquals(scored, res, 2, 0, 1e-4); + assertColsEquals(scored, res, 2, 0, 1e-3); } finally { fr.delete(); bgFr.delete(); @@ -109,7 +109,7 @@ public void testClassificationOriginalSHAP() { Val val = Rapids.exec("(sumaxis " + contribs._key + " 0 1)"); assertTrue(val instanceof ValFrame); res = val.getFrame(); - assertColsEquals(scored, res, 2, 0, 1e-4); + assertColsEquals(scored, res, 2, 0, 1e-3); } finally { fr.delete(); bgFr.delete(); @@ -152,7 +152,7 @@ public void testRegressionCompactSHAP() { Val val = Rapids.exec("(sumaxis " + contribs._key + " 0 1)"); assertTrue(val instanceof ValFrame); res = val.getFrame(); - assertColsEquals(scored, res, 0, 0, 1e-4); + assertColsEquals(scored, res, 0, 0, 1e-3); } finally { fr.delete(); bgFr.delete(); @@ -194,7 +194,7 @@ public void testRegressionOriginalSHAP() { Val val = Rapids.exec("(sumaxis " + contribs._key + " 0 1)"); assertTrue(val instanceof ValFrame); res = val.getFrame(); - assertColsEquals(scored, res, 0, 0, 1e-4); + assertColsEquals(scored, res, 0, 0, 1e-3); } finally { fr.delete(); bgFr.delete();