From e4d1c2830cd29c5bed11bef2bf9479bb7475778e Mon Sep 17 00:00:00 2001 From: Adam Pocock Date: Sun, 26 Sep 2021 22:21:20 -0400 Subject: [PATCH] Relaxing the similarity test in FMRegressionModel as the standardisation is quite different in floats. --- .../java/org/tribuo/regression/sgd/fm/TestFMRegression.java | 2 +- 1 file changed, 1 insertion(+), 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 index 33a45b4c5..09e8e8fd1 100644 --- 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 tribuo = nativePredictions.get(i); Prediction 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();