c

com.johnsnowlabs.ml.onnx

OnnxMedicalEncoderDecoderT5

class OnnxMedicalEncoderDecoderT5 extends T5MedicalEncoderDecoder

Linear Supertypes
T5MedicalEncoderDecoder, MedicalEncoderDecoderModel, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. OnnxMedicalEncoderDecoderT5
  2. T5MedicalEncoderDecoder
  3. MedicalEncoderDecoderModel
  4. AnyRef
  5. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

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

Type Members

  1. class DecoderProcessor extends AnyRef
    Definition Classes
    T5MedicalEncoderDecoder

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
    OnnxMedicalEncoderDecoderT5 → T5MedicalEncoderDecoder
  5. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  6. def calcBannedNgramTokens(prevInputIds: Seq[Array[Int]], numHypos: Int, noRepeatNgramSize: Int, curLen: Int): Array[Array[Int]]
    Attributes
    protected
    Definition Classes
    T5MedicalEncoderDecoder
  7. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  8. def decode(sentences: Array[Array[Int]]): Seq[String]
    Attributes
    protected
    Definition Classes
    T5MedicalEncoderDecoder
  9. lazy val decoderCacheInputKeys: Array[String]
  10. lazy val decoderCacheOutputKeys: Array[String]
  11. def encode(sentences: Seq[Annotation], task: String): Seq[Array[Int]]
    Attributes
    protected
    Definition Classes
    T5MedicalEncoderDecoder
  12. def encodeS(sentences: Seq[String], task: String): Seq[Array[Int]]
    Attributes
    protected
    Definition Classes
    T5MedicalEncoderDecoder
  13. lazy val encoderCacheInputKeys: Array[String]
  14. lazy val encoderCacheOutputKeys: Array[String]
  15. val eosTokenId: Int
    Attributes
    protected
    Definition Classes
    T5MedicalEncoderDecoder
  16. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  17. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  18. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  19. def generate(prompts: Seq[Annotation], batchSize: Int, maxNewTokens: Int, maxContextLength: Int, doSample: Boolean, topK: Int, randomSeed: Option[Int], ignoreTokenIds: Array[Int], isCaseSensitive: Boolean, stopAtEos: Boolean, noRepeatNgramSize: Int): Seq[Annotation]
    Definition Classes
    T5MedicalEncoderDecoder → MedicalEncoderDecoderModel
  20. def generateCacheKeys(component: String, state: String): Array[String]
  21. def generateNoBeamSearch(inputIds: Seq[Array[Int]], encoderStateTensors: OnnxTensor, encoderAttentionMaskTensors: OnnxTensor, maxNewTokens: Int, maxTextLength: Int, doSample: Boolean, topK: Int, vocabSize: Int, randomSeed: Option[Int], ignoreTokenIds: Array[Int] = Array(), stopAtEos: Boolean, noRepeatNgramSize: Int): (Array[Array[Int]], Array[Float])
  22. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  23. def getGeneratedNgrams(prevInputIds: Seq[Array[Int]], generatedNgrams: Array[Map[IndexedSeq[Int], List[Int]]], hypoIdx: Int, curLen: Int, noRepeatNgramSize: Int): Array[Int]
    Attributes
    protected
    Definition Classes
    T5MedicalEncoderDecoder
  24. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  25. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  26. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  27. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  28. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  29. val numAttnHeads: Int
    Attributes
    protected
  30. val numLayers: Int
    Attributes
    protected
  31. val onnxDecoder: OnnxWrapper
  32. val onnxEncoder: OnnxWrapper
  33. val paddingTokenId: Int
    Attributes
    protected
    Definition Classes
    T5MedicalEncoderDecoder
  34. val pieceSize: Int
    Attributes
    protected
    Definition Classes
    T5MedicalEncoderDecoder
  35. def predict(sentences: Seq[Annotation], task: String, batchSize: Int, maxNewTokens: Int, maxTextLength: Int, doSample: Boolean, topK: Int, randomSeed: Option[Int] = None, ignoreTokenIds: Array[Int] = Array(), isCaseSensitive: Boolean, stopAtEos: Boolean, noRepeatNgramSize: Int, refineSummary: Boolean = false, refineSummaryTargetLength: Int = 100, refineChunkSize: Int = 512, refineMaxAttempts: Int = 3): Seq[Annotation]
    Definition Classes
    T5MedicalEncoderDecoder → MedicalEncoderDecoderModel
  36. def predictQuestions(questionAndContexts: Seq[(Annotation, Annotation)], batchSize: Int, maxNewTokens: Int, maxContextLength: Int, doSample: Boolean, topK: Int, questionPrompt: String, randomSeed: Option[Int] = None, ignoreTokenIds: Array[Int] = Array(), isCaseSensitive: Boolean, questionAnswerTerminals: Array[Int], skipLastToken: Boolean, useCache: Boolean, noRepeatNgramSize: Int): Seq[Annotation]
    Definition Classes
    MedicalEncoderDecoderModel
  37. def refinedPredict(batch: Seq[String], task: String, maxNewTokens: Int, maxTextLength: Int, doSample: Boolean, topK: Int, randomSeed: Option[Int] = None, ignoreTokenIds: Array[Int] = Array(), stopAtEos: Boolean, noRepeatNgramSize: Int, refineSummaryTargetLength: Int = 100, refineChunkSize: Int = 512, sumBelowTargetLength: Boolean = true): (Seq[String], Seq[Float])
    Definition Classes
    T5MedicalEncoderDecoder
  38. def sessionWarmup(): Unit
    Definition Classes
    T5MedicalEncoderDecoder → MedicalEncoderDecoderModel
  39. def setTensorByIndicesToValue(prevInputIds: Array[Float], indices: IndexedSeq[Boolean], value: Float): Array[Float]
    Attributes
    protected
    Definition Classes
    T5MedicalEncoderDecoder
  40. val spp: SentencePieceWrapper
    Definition Classes
    OnnxMedicalEncoderDecoderT5 → T5MedicalEncoderDecoder
  41. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  42. def tag(batch: Seq[Array[Int]], maxNewTokens: Int, maxTextLength: Int, doSample: Boolean, topK: Int, randomSeed: Option[Int], ignoreTokenIds: Array[Int] = Array(), stopAtEos: Boolean, noRepeatNgramSize: Int): (Array[Array[Int]], Array[Float])
    Definition Classes
    OnnxMedicalEncoderDecoderT5 → T5MedicalEncoderDecoder
  43. def toString(): String
    Definition Classes
    AnyRef → Any
  44. val vocabSize: Int
    Attributes
    protected
    Definition Classes
    T5MedicalEncoderDecoder
  45. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  46. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  47. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()

Inherited from T5MedicalEncoderDecoder

Inherited from MedicalEncoderDecoderModel

Inherited from AnyRef

Inherited from Any

Ungrouped