diff options
author | Lester Solbakken <lesters@oath.com> | 2021-08-31 14:40:50 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2021-08-31 14:40:50 +0200 |
commit | 522578bb7392c95ac0c8ef2b599f65d6d79df987 (patch) | |
tree | 6da6b89b6a2ec1a06afed512c8498f16033b7001 /model-evaluation/src | |
parent | 28ed1156289bdb5d6e874c3a84a86f46789dd440 (diff) |
Parse unbound tensors in short form
Diffstat (limited to 'model-evaluation/src')
-rw-r--r-- | model-evaluation/src/test/java/ai/vespa/models/handler/ModelsEvaluationHandlerTest.java | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/model-evaluation/src/test/java/ai/vespa/models/handler/ModelsEvaluationHandlerTest.java b/model-evaluation/src/test/java/ai/vespa/models/handler/ModelsEvaluationHandlerTest.java index 5b1c7a478b1..df89919a76e 100644 --- a/model-evaluation/src/test/java/ai/vespa/models/handler/ModelsEvaluationHandlerTest.java +++ b/model-evaluation/src/test/java/ai/vespa/models/handler/ModelsEvaluationHandlerTest.java @@ -174,6 +174,15 @@ public class ModelsEvaluationHandlerTest { } @Test + public void testMnistSoftmaxEvaluateSpecificFunctionWithBindingsShortForm() { + Map<String, String> properties = new HashMap<>(); + properties.put("Placeholder", inputTensorShortForm()); + String url = "http://localhost/model-evaluation/v1/mnist_softmax/default.add/eval"; + String expected = "{\"cells\":[{\"address\":{\"d0\":\"0\",\"d1\":\"0\"},\"value\":-0.3546536862850189},{\"address\":{\"d0\":\"0\",\"d1\":\"1\"},\"value\":0.3759574592113495},{\"address\":{\"d0\":\"0\",\"d1\":\"2\"},\"value\":0.06054411828517914},{\"address\":{\"d0\":\"0\",\"d1\":\"3\"},\"value\":-0.251544713973999},{\"address\":{\"d0\":\"0\",\"d1\":\"4\"},\"value\":0.017951013520359993},{\"address\":{\"d0\":\"0\",\"d1\":\"5\"},\"value\":1.2899067401885986},{\"address\":{\"d0\":\"0\",\"d1\":\"6\"},\"value\":-0.10389615595340729},{\"address\":{\"d0\":\"0\",\"d1\":\"7\"},\"value\":0.6367976665496826},{\"address\":{\"d0\":\"0\",\"d1\":\"8\"},\"value\":-1.4136744737625122},{\"address\":{\"d0\":\"0\",\"d1\":\"9\"},\"value\":-0.2573896050453186}]}"; + handler.assertResponse(url, properties, 200, expected); + } + + @Test public void testMnistSavedDetails() { String url = "http://localhost:8080/model-evaluation/v1/mnist_saved"; String expected = "{\"model\":\"mnist_saved\",\"functions\":[{\"function\":\"serving_default.y\",\"info\":\"http://localhost:8080/model-evaluation/v1/mnist_saved/serving_default.y\",\"eval\":\"http://localhost:8080/model-evaluation/v1/mnist_saved/serving_default.y/eval\",\"arguments\":[{\"name\":\"input\",\"type\":\"tensor(d0[],d1[784])\"}]}]}"; @@ -224,4 +233,17 @@ public class ModelsEvaluationHandlerTest { return b.build().toString(); } + private String inputTensorShortForm() { + StringBuilder sb = new StringBuilder(); + sb.append("[["); + for (int i = 0; i < 784; i++) { + sb.append("0.0"); + if (i < 783) { + sb.append(","); + } + } + sb.append("]]"); + return sb.toString(); + } + } |