From e2d1926da7646ed92c177c6bc960893e2690a3bc Mon Sep 17 00:00:00 2001 From: Jon Bratseth Date: Mon, 10 Sep 2018 14:49:23 +0200 Subject: Test mnist model --- .../src/test/java/ai/vespa/models/evaluation/MlModelsImportingTest.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'model-evaluation/src') diff --git a/model-evaluation/src/test/java/ai/vespa/models/evaluation/MlModelsImportingTest.java b/model-evaluation/src/test/java/ai/vespa/models/evaluation/MlModelsImportingTest.java index f236bbd4467..6e55c0c9a53 100644 --- a/model-evaluation/src/test/java/ai/vespa/models/evaluation/MlModelsImportingTest.java +++ b/model-evaluation/src/test/java/ai/vespa/models/evaluation/MlModelsImportingTest.java @@ -67,7 +67,7 @@ public class MlModelsImportingTest { tfMnist); FunctionEvaluator evaluator = tfMnist.evaluatorOf("serving_default"); // TODO: Macro is offered as an alternative output currently, so need to specify argument assertEquals("constant(mnist_saved_dnn_hidden1_bias_read), constant(mnist_saved_dnn_hidden1_weights_read), constant(mnist_saved_dnn_hidden2_bias_read), constant(mnist_saved_dnn_hidden2_weights_read), constant(mnist_saved_dnn_outputs_bias_read), constant(mnist_saved_dnn_outputs_weights_read), input, rankingExpression(imported_ml_macro_mnist_saved_dnn_hidden1_add)", evaluator.context().names().stream().sorted().collect(Collectors.joining(", "))); - assertEquals(-0.714629131972222, evaluator.evaluate().sum().asDouble(), delta); // TODO: Verify in TF native + assertEquals(-0.714629131972222, evaluator.evaluate().sum().asDouble(), delta); } } -- cgit v1.2.3