diff options
author | Jon Bratseth <bratseth@gmail.com> | 2021-12-16 18:35:11 +0100 |
---|---|---|
committer | Jon Bratseth <bratseth@gmail.com> | 2021-12-16 18:35:11 +0100 |
commit | 767cb63af0f530605180f5438767406e1db27520 (patch) | |
tree | c0ea9e8ec4ded2dea6064a45334e6f8a1408f7b8 /linguistics/src/main/java/com/yahoo/language/process/Embedder.java | |
parent | 1eefb9854bcd7ca264889239b32e7fc8c8830672 (diff) |
Add a BERT embedder
Diffstat (limited to 'linguistics/src/main/java/com/yahoo/language/process/Embedder.java')
-rw-r--r-- | linguistics/src/main/java/com/yahoo/language/process/Embedder.java | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/linguistics/src/main/java/com/yahoo/language/process/Embedder.java b/linguistics/src/main/java/com/yahoo/language/process/Embedder.java index 17ee0419cea..dd9c3847314 100644 --- a/linguistics/src/main/java/com/yahoo/language/process/Embedder.java +++ b/linguistics/src/main/java/com/yahoo/language/process/Embedder.java @@ -3,6 +3,7 @@ package com.yahoo.language.process; import com.yahoo.language.Language; import com.yahoo.tensor.Tensor; +import com.yahoo.tensor.TensorAddress; import com.yahoo.tensor.TensorType; import java.util.List; @@ -34,7 +35,7 @@ public interface Embedder { * @param text the text to embed * @param context the context which may influence an embedder's behavior * @param tensorType the type of the tensor to be returned - * @return the tensor embedding of the text, as the spoecified tensor type + * @return the tensor embedding of the text, as the specified tensor type * @throws IllegalArgumentException if the language or tensor type is not supported by this embedder */ Tensor embed(String text, Context context, TensorType tensorType); @@ -69,7 +70,7 @@ public interface Embedder { /** * Sets the name of the recipient of this tensor. * - * This iseither a query feature name + * This is either a query feature name * ("query(feature)"), or a schema and field name concatenated by a dot ("schema.field"). */ public Context setDestination(String destination) { |