diff options
author | Jon Bratseth <bratseth@oath.com> | 2018-08-28 15:30:51 +0200 |
---|---|---|
committer | Jon Bratseth <bratseth@oath.com> | 2018-08-28 15:30:51 +0200 |
commit | a97182c76ce577535353efcee9ccab0c51ece200 (patch) | |
tree | c5ba62adb32851b8e7605a8461e33ce889622661 /model-evaluation | |
parent | d4667e29011ea3b39b188438199ec9a0400b70ed (diff) |
API improvements
Diffstat (limited to 'model-evaluation')
-rw-r--r-- | model-evaluation/src/main/java/ai/vespa/models/evaluation/ModelsEvaluator.java | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/model-evaluation/src/main/java/ai/vespa/models/evaluation/ModelsEvaluator.java b/model-evaluation/src/main/java/ai/vespa/models/evaluation/ModelsEvaluator.java index 50e83fd1a1e..682a7fa057c 100644 --- a/model-evaluation/src/main/java/ai/vespa/models/evaluation/ModelsEvaluator.java +++ b/model-evaluation/src/main/java/ai/vespa/models/evaluation/ModelsEvaluator.java @@ -31,14 +31,16 @@ public class ModelsEvaluator extends AbstractComponent { /** * Returns a function which can be used to evaluate the given function in the given model * + * @param modelName the name of the model + * @param names the 0-2 name components identifying the output to compute * @throws IllegalArgumentException if the function or model is not present */ - public FunctionEvaluator evaluatorOf(String modelName, String functionName) { - return requireModel(modelName).evaluatorOf(functionName); + public FunctionEvaluator evaluatorOf(String modelName, String ... names) { + return requireModel(modelName).evaluatorOf(names); } /** Returns the given model, or throws a IllegalArgumentException if it does not exist */ - Model requireModel(String name) { + public Model requireModel(String name) { Model model = models.get(name); if (model == null) throw new IllegalArgumentException("No model named '" + name + ". Available models: " + |