Relaxing the similarity test in FMRegressionModel as the standardisat…

…ion is quite different in floats.
oracle · Oct 1, 2021 · e4d1c28 · e4d1c28
1 parent 4ea0252
commit e4d1c28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Regression/SGD/src/test/java/org/tribuo/regression/sgd/fm/TestFMRegression.java b/Regression/SGD/src/test/java/org/tribuo/regression/sgd/fm/TestFMRegression.java
@@ -171,7 +171,7 @@ public void testOnnxSerialization() throws IOException, OrtException {
                 Prediction<Regressor> tribuo = nativePredictions.get(i);
                 Prediction<Regressor> external = onnxPredictions.get(i);
                 assertArrayEquals(tribuo.getOutput().getNames(),external.getOutput().getNames());
-                assertArrayEquals(tribuo.getOutput().getValues(),external.getOutput().getValues(),1e-5);
+                assertArrayEquals(tribuo.getOutput().getValues(),external.getOutput().getValues(),1e-4);
             }
 
             onnxModel.close();