Make DeepSHAP tests more stable (#15826)

h2oai · Oct 17, 2023 · 29d84fc · 29d84fc
1 parent 24298f6
commit 29d84fc
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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();