Packages

c

com.johnsnowlabs.ml.ai.seq2seq

OnnxT5EncoderDecoder

class OnnxT5EncoderDecoder extends T5EncoderDecoder

Linear Supertypes
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. OnnxT5EncoderDecoder
  2. T5EncoderDecoder
  3. AnyRef
  4. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new OnnxT5EncoderDecoder(onnxEncoder: OnnxWrapper, onnxDecoder: OnnxWrapper, spp: SentencePieceWrapper, additionalTokens: Map[Int, String] = Map())

Value Members

  1. final def !=(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int
    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  4. val additionalTokens: Map[Int, String]
    Definition Classes
    OnnxT5EncoderDecoderT5EncoderDecoder
  5. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  6. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  7. def decode(sentences: Array[Array[Int]]): Seq[String]
    Attributes
    protected
    Definition Classes
    T5EncoderDecoder
  8. lazy val decoderCacheInputKeys: Array[String]
  9. lazy val decoderCacheOutputKeys: Array[String]
  10. def encode(sentences: Seq[Annotation], task: String): Seq[Array[Int]]
    Attributes
    protected
    Definition Classes
    T5EncoderDecoder
  11. lazy val encoderCacheInputKeys: Array[String]
  12. lazy val encoderCacheOutputKeys: Array[String]
  13. val eosTokenId: Int
    Attributes
    protected
    Definition Classes
    T5EncoderDecoder
  14. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  15. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  16. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  17. def generate(prompts: Seq[Annotation], batchSize: Int, maxNewTokens: Int, maxContextLength: Int, doSample: Boolean, topK: Int, topP: Double, temperature: Double, randomSeed: Option[Long], ignoreTokenIds: Array[Int], isCaseSensitive: Boolean, stopAtEos: Boolean, noRepeatNgramSize: Int, repetitionPenalty: Double): Seq[Annotation]
    Definition Classes
    T5EncoderDecoder
  18. def generateCacheKeys(component: String, state: String): Array[String]
  19. def generateNoBeamSearch(inputIds: Seq[Array[Int]], encoderStateTensors: OnnxTensor, encoderAttentionMaskTensors: OnnxTensor, maxNewTokens: Int, maxTextLength: Int, doSample: Boolean, topK: Int, topP: Double, temperature: Double, vocabSize: Int, randomSeed: Option[Long], ignoreTokenIds: Array[Int] = Array(), stopAtEos: Boolean, noRepeatNgramSize: Int, repetitionPenalty: Double): Array[Array[Int]]
  20. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  21. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  22. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  23. val logger: Logger
    Attributes
    protected
  24. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  25. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  26. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  27. val numAttnHeads: Int
    Attributes
    protected
  28. val numLayers: Int
    Attributes
    protected
  29. val onnxDecoder: OnnxWrapper
  30. val onnxEncoder: OnnxWrapper
  31. val paddingTokenId: Int
    Attributes
    protected
    Definition Classes
    T5EncoderDecoder
  32. val pieceSize: Int
    Attributes
    protected
    Definition Classes
    T5EncoderDecoder
  33. def predict(sentences: Seq[Annotation], task: String, batchSize: Int, maxNewTokens: Int, maxTextLength: Int, doSample: Boolean, topK: Int, topP: Double, temperature: Double, randomSeed: Option[Long] = None, ignoreTokenIds: Array[Int] = Array(), isCaseSensitive: Boolean, stopAtEos: Boolean, noRepeatNgramSize: Int, repetitionPenalty: Double): Seq[Annotation]
    Definition Classes
    T5EncoderDecoder
  34. def sessionWarmup(): Unit
    Definition Classes
    T5EncoderDecoder
  35. val spp: SentencePieceWrapper
    Definition Classes
    OnnxT5EncoderDecoderT5EncoderDecoder
  36. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  37. def tag(batch: Seq[Array[Int]], maxNewTokens: Int, maxTextLength: Int, doSample: Boolean, topK: Int, topP: Double, temperature: Double, noRepeatNgramSize: Int, repetitionPenalty: Double, randomSeed: Option[Long], ignoreTokenIds: Array[Int] = Array(), stopAtEos: Boolean): Array[Array[Int]]
    Definition Classes
    OnnxT5EncoderDecoderT5EncoderDecoder
  38. def toString(): String
    Definition Classes
    AnyRef → Any
  39. val vocabSize: Int
    Attributes
    protected
    Definition Classes
    T5EncoderDecoder
  40. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  41. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  42. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()

Inherited from T5EncoderDecoder

Inherited from AnyRef

Inherited from Any

Ungrouped