aboutsummaryrefslogtreecommitdiffstats
path: root/linguistics
diff options
context:
space:
mode:
authorJon Bratseth <bratseth@gmail.com>2021-12-16 18:35:11 +0100
committerJon Bratseth <bratseth@gmail.com>2021-12-16 18:35:11 +0100
commit767cb63af0f530605180f5438767406e1db27520 (patch)
treec0ea9e8ec4ded2dea6064a45334e6f8a1408f7b8 /linguistics
parent1eefb9854bcd7ca264889239b32e7fc8c8830672 (diff)
Add a BERT embedder
Diffstat (limited to 'linguistics')
-rw-r--r--linguistics/src/main/java/com/yahoo/language/process/Embedder.java5
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) {