diff options
Diffstat (limited to 'model-evaluation')
-rw-r--r-- | model-evaluation/src/main/java/ai/vespa/models/evaluation/Model.java | 2 | ||||
-rw-r--r-- | model-evaluation/src/main/java/ai/vespa/models/evaluation/ModelsEvaluator.java | 8 |
2 files changed, 7 insertions, 3 deletions
diff --git a/model-evaluation/src/main/java/ai/vespa/models/evaluation/Model.java b/model-evaluation/src/main/java/ai/vespa/models/evaluation/Model.java index 593387e08bf..95eb923786d 100644 --- a/model-evaluation/src/main/java/ai/vespa/models/evaluation/Model.java +++ b/model-evaluation/src/main/java/ai/vespa/models/evaluation/Model.java @@ -126,6 +126,8 @@ public class Model { * @param names the names identifying the function - this can be from 0 to 2, specifying function or "signature" * name, and "output", respectively. Names which are unnecessary to determine the desired function * uniquely (e.g if there is just one function or output) can be omitted. + * A two-component name can alternatively be specified as a single argument with components separated + * by dot. * @throws IllegalArgumentException if the function is not present, or not uniquely identified by the names given */ public FunctionEvaluator evaluatorOf(String ... names) { // TODO: Parameter overloading? 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 ca5cdc3cc56..dacf20b7ef2 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: " + |