Class

com.johnsnowlabs.ml.tensorflow

TensorflowT5

Related Doc: package tensorflow

Permalink

class TensorflowT5 extends Serializable

This class is used to run T5 model for For Sequence Batches of WordpieceTokenizedSentence. Input for this model must be tokenized with a SentencePieceModel,

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

Instance Constructors

  1. new TensorflowT5(tensorflow: TensorflowWrapper, spp: SentencePieceWrapper, configProtoBytes: Option[Array[Byte]] = None)

    Permalink

    tensorflow

    Albert Model wrapper with TensorFlowWrapper

    spp

    Albert SentencePiece model with SentencePieceWrapper

    configProtoBytes

    Configuration for TensorFlow session

Value Members

  1. final def !=(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any

  2. final def ##(): Int

    Permalink
    Definition Classes
    AnyRef → Any

  3. final def ==(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any

  4. final def asInstanceOf[T0]: T0

    Permalink
    Definition Classes
    Any

  5. def clone(): AnyRef

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  6. def createNextTokenLogitsPenalties(inputIds: Seq[Array[Long]], logits: Array[Array[Float]], repetitionPenalty: Double): Array[Array[Float]]

    Permalink

  7. def decode(sentences: Array[Array[Long]]): Seq[String]

    Permalink

  8. def encode(sentences: Seq[Annotation], task: String): Seq[Array[Long]]

    Permalink

  9. final def eq(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef

  10. def equals(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any

  11. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )

  12. def generateNoBeamSearch(inputIds: Seq[Array[Long]], decoderEncoderStateTensors: Tensor, encoderAttentionMaskTensors: Tensor, maxOutputLength: Int, minOutputLength: Int, doSample: Boolean, temperature: Double, topK: Int, topP: Double, repetitionPenalty: Double, noRepeatNgramSize: Int, batch_size: Int, vocab_size: Int, randomSeed: Option[Long], session: Session): Array[Array[Long]]

    Permalink

  13. def generateSeq2Seq(sentences: Seq[Annotation], batchSize: Int, minOutputLength: Int, maxOutputLength: Int, doSample: Boolean, temperature: Double, topK: Int, topP: Double, repetitionPenalty: Double, noRepeatNgramSize: Int, task: String, randomSeed: Option[Long] = None): Seq[Annotation]

    Permalink

  14. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any

  15. def getGeneratedNgrams(prevInputIds: Seq[Array[Long]], generatedNgrams: Array[Map[IndexedSeq[Long], List[Long]]], hypoIdx: Int, curLen: Int, noRepeatNgramSize: Int): Array[Long]

    Permalink

  16. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any

  17. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any

  18. final def ne(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef

  19. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef

  20. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef

  21. def process(batch: Seq[Array[Long]], minOutputLength: Int, maxOutputLength: Int, doSample: Boolean, temperature: Double, topK: Int, topP: Double, repetitionPenalty: Double, noRepeatNgramSize: Int, randomSeed: Option[Long]): Array[Array[Long]]

    Permalink

  22. val spp: SentencePieceWrapper

    Permalink

    Albert SentencePiece model with SentencePieceWrapper

  23. final def synchronized[T0](arg0: ⇒ T0): T0

    Permalink
    Definition Classes
    AnyRef

  24. val tensorflow: TensorflowWrapper

    Permalink

    Albert Model wrapper with TensorFlowWrapper

  25. def toString(): String

    Permalink
    Definition Classes
    AnyRef → Any

  26. final def wait(): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  27. final def wait(arg0: Long, arg1: Int): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  28. final def wait(arg0: Long): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from Serializable

Inherited from Serializable

Inherited from AnyRef

Inherited from Any

Ungrouped