diff options
Diffstat (limited to 'model-integration')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Softmax.java | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Softmax.java b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Softmax.java index 8db237d15e4..cdacbe1656a 100644 --- a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Softmax.java +++ b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Softmax.java @@ -6,6 +6,11 @@ import com.yahoo.tensor.functions.TensorFunction; import java.util.List; +/** + * Convert imported 'softmax' operation to the Vespa softmax ranking function. + * + * @author lesters + */ public class Softmax extends IntermediateOperation { public Softmax(String modelName, String nodeName, List<IntermediateOperation> inputs) { |