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, ignoreTokenIds: Array[Int] = Array()): Array[Array[Long]]

    Permalink
  13. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any
  14. def getGeneratedNgrams(prevInputIds: Seq[Array[Long]], generatedNgrams: Array[Map[IndexedSeq[Long], List[Long]]], hypoIdx: Int, curLen: Int, noRepeatNgramSize: Int): Array[Long]

    Permalink
  15. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  16. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any
  17. final def ne(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  18. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef
  19. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef
  20. def predict(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, ignoreTokenIds: Array[Int] = Array()): Seq[Annotation]

    Permalink
  21. val spp: SentencePieceWrapper

    Permalink

    Albert SentencePiece model with SentencePieceWrapper

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

    Permalink
    Definition Classes
    AnyRef
  23. def tag(batch: Seq[Array[Long]], minOutputLength: Int, maxOutputLength: Int, doSample: Boolean, temperature: Double, topK: Int, topP: Double, repetitionPenalty: Double, noRepeatNgramSize: Int, randomSeed: Option[Long], ignoreTokenIds: Array[Int] = Array()): Array[Array[Long]]

    Permalink
  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