class Text2SQL extends MedicalTextGenerator

Linear Supertypes
MedicalTextGenerator, CheckLicense, HasEngine, WriteSentencePieceModel, WriteOnnxModel, WriteTensorflowModel, HasBatchedAnnotate[MedicalQuestionAnswering], HasCaseSensitiveProperties, GPTGenerationParams, AnnotatorModel[MedicalQuestionAnswering], CanBeLazy, RawAnnotator[MedicalQuestionAnswering], HasOutputAnnotationCol, HasInputAnnotationCols, HasOutputAnnotatorType, ParamsAndFeaturesWritable, HasFeatures, DefaultParamsWritable, MLWritable, Model[MedicalQuestionAnswering], Transformer, PipelineStage, Logging, Params, Serializable, Serializable, Identifiable, AnyRef, Any
Ordering
  1. Grouped
  2. Alphabetic
  3. By Inheritance
Inherited
  1. Text2SQL
  2. MedicalTextGenerator
  3. CheckLicense
  4. HasEngine
  5. WriteSentencePieceModel
  6. WriteOnnxModel
  7. WriteTensorflowModel
  8. HasBatchedAnnotate
  9. HasCaseSensitiveProperties
  10. GPTGenerationParams
  11. AnnotatorModel
  12. CanBeLazy
  13. RawAnnotator
  14. HasOutputAnnotationCol
  15. HasInputAnnotationCols
  16. HasOutputAnnotatorType
  17. ParamsAndFeaturesWritable
  18. HasFeatures
  19. DefaultParamsWritable
  20. MLWritable
  21. Model
  22. Transformer
  23. PipelineStage
  24. Logging
  25. Params
  26. Serializable
  27. Serializable
  28. Identifiable
  29. AnyRef
  30. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new Text2SQL()
  2. new Text2SQL(uid: String)

Type Members

  1. type AnnotationContent = Seq[Row]
    Attributes
    protected
    Definition Classes
    AnnotatorModel
  2. type AnnotatorType = String
    Definition Classes
    HasOutputAnnotatorType

Value Members

  1. final def !=(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int
    Definition Classes
    AnyRef → Any
  3. final def $[T](param: Param[T]): T
    Attributes
    protected
    Definition Classes
    Params
  4. def $$[T](feature: StructFeature[T]): T
    Attributes
    protected
    Definition Classes
    HasFeatures
  5. def $$[K, V](feature: MapFeature[K, V]): Map[K, V]
    Attributes
    protected
    Definition Classes
    HasFeatures
  6. def $$[T](feature: SetFeature[T]): Set[T]
    Attributes
    protected
    Definition Classes
    HasFeatures
  7. def $$[T](feature: ArrayFeature[T]): Array[T]
    Attributes
    protected
    Definition Classes
    HasFeatures
  8. final def ==(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  9. val DOCUMENT_VARIABLE_NAME: String
    Attributes
    protected
    Definition Classes
    MedicalTextGenerator
  10. val QUESTION_VARIABLE_NAME: String
    Attributes
    protected
  11. val TABLES_VARIABLE_NAME: String
    Attributes
    protected
  12. val TABLES_WITH_COLUMNS_VARIABLE_NAME: String
    Attributes
    protected
  13. val TABLE_COLUMNS_VARIABLE_NAME: String
    Attributes
    protected
  14. val TABLE_NAME_VARIABLE_NAME: String
    Attributes
    protected
  15. def _transform(dataset: Dataset[_], recursivePipeline: Option[PipelineModel]): DataFrame
    Attributes
    protected
    Definition Classes
    AnnotatorModel
  16. val additionalTokens: MapFeature[Int, String]

    Additional tokens

    Additional tokens

    Definition Classes
    MedicalTextGenerator
  17. def afterAnnotate(dataset: DataFrame): DataFrame
    Attributes
    protected
    Definition Classes
    AnnotatorModel
  18. def applyVariablesToPrompt(prompt: String, question: String): String
  19. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  20. def batchAnnotate(batchedAnnotations: Seq[Array[Annotation]]): Seq[Seq[Annotation]]

    takes a document and annotations and produces new annotations of this annotator's annotation type

    takes a document and annotations and produces new annotations of this annotator's annotation type

    batchedAnnotations

    Annotations that correspond to inputAnnotationCols generated by previous annotators if any

    returns

    any number of annotations processed for every input annotation. Not necessary one to one relationship

    Definition Classes
    Text2SQLMedicalTextGenerator → HasBatchedAnnotate
  21. def batchProcess(rows: Iterator[_]): Iterator[Row]
    Definition Classes
    HasBatchedAnnotate
  22. val batchSize: IntParam
    Definition Classes
    HasBatchedAnnotate
  23. def batchedAnnotateWithoutPromptTemplate(batchedAnnotations: Seq[Array[Annotation]]): Seq[Seq[Annotation]]
    Definition Classes
    MedicalTextGenerator
  24. def beforeAnnotate(dataset: Dataset[_]): Dataset[_]
    Attributes
    protected
    Definition Classes
    AnnotatorModel
  25. val caseSensitive: BooleanParam
    Definition Classes
    HasCaseSensitiveProperties
  26. final def checkSchema(schema: StructType, inputAnnotatorType: String): Boolean
    Attributes
    protected
    Definition Classes
    HasInputAnnotationCols
  27. def checkValidEnvironment(spark: Option[SparkSession], scopes: Seq[String]): Unit
    Definition Classes
    CheckLicense
  28. def checkValidScope(scope: String): Unit
    Definition Classes
    CheckLicense
  29. def checkValidScopeAndEnvironment(scope: String, spark: Option[SparkSession], checkLp: Boolean): Unit
    Definition Classes
    CheckLicense
  30. def checkValidScopesAndEnvironment(scopes: Seq[String], spark: Option[SparkSession], checkLp: Boolean): Unit
    Definition Classes
    CheckLicense
  31. final def clear(param: Param[_]): Text2SQL.this.type
    Definition Classes
    Params
  32. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  33. val configProtoBytes: IntArrayParam

    ConfigProto from tensorflow, serialized into byte array.

    ConfigProto from tensorflow, serialized into byte array. Get with config_proto.SerializeToString()

    Definition Classes
    MedicalTextGenerator
  34. def copy(extra: ParamMap): MedicalQuestionAnswering
    Definition Classes
    RawAnnotator → Model → Transformer → PipelineStage → Params
  35. def copyValues[T <: Params](to: T, extra: ParamMap): T
    Attributes
    protected
    Definition Classes
    Params
  36. val customPrompt: Param[String]

    Custom model prompt

    Custom model prompt

    Definition Classes
    MedicalTextGenerator
  37. final def defaultCopy[T <: Params](extra: ParamMap): T
    Attributes
    protected
    Definition Classes
    Params
  38. val doSample: BooleanParam

    Whether or not to use sampling, use greedy decoding otherwise (Default: false)

    Whether or not to use sampling, use greedy decoding otherwise (Default: false)

    Definition Classes
    GPTGenerationParams
  39. val engine: Param[String]
    Definition Classes
    HasEngine
  40. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  41. def equals(arg0: Any): Boolean
    Definition Classes
    AnyRef → Any
  42. def explainParam(param: Param[_]): String
    Definition Classes
    Params
  43. def explainParams(): String
    Definition Classes
    Params
  44. def extraValidate(structType: StructType): Boolean
    Attributes
    protected
    Definition Classes
    RawAnnotator
  45. def extraValidateMsg: String
    Attributes
    protected
    Definition Classes
    RawAnnotator
  46. final def extractParamMap(): ParamMap
    Definition Classes
    Params
  47. final def extractParamMap(extra: ParamMap): ParamMap
    Definition Classes
    Params
  48. val features: ArrayBuffer[Feature[_, _, _]]
    Definition Classes
    HasFeatures
  49. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  50. def generateSchemaVariables(): Map[String, String]
  51. def get[T](feature: StructFeature[T]): Option[T]
    Attributes
    protected
    Definition Classes
    HasFeatures
  52. def get[K, V](feature: MapFeature[K, V]): Option[Map[K, V]]
    Attributes
    protected
    Definition Classes
    HasFeatures
  53. def get[T](feature: SetFeature[T]): Option[Set[T]]
    Attributes
    protected
    Definition Classes
    HasFeatures
  54. def get[T](feature: ArrayFeature[T]): Option[Array[T]]
    Attributes
    protected
    Definition Classes
    HasFeatures
  55. final def get[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  56. def getAdditionalTokens: Map[Int, String]

    Get additional tokens

    Get additional tokens

    Definition Classes
    MedicalTextGenerator
  57. def getAdditionalTokensStr: String

    Get additional tokens in string format

    Get additional tokens in string format

    Definition Classes
    MedicalTextGenerator
  58. def getBatchSize: Int
    Definition Classes
    HasBatchedAnnotate
  59. def getCaseSensitive: Boolean
    Definition Classes
    HasCaseSensitiveProperties
  60. final def getClass(): Class[_]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  61. def getConfigProtoBytes: Option[Array[Byte]]

    Definition Classes
    MedicalTextGenerator
  62. def getCustomPrompt: String

    Custom model prompt

    Custom model prompt

    Attributes
    protected
    Definition Classes
    MedicalTextGenerator
  63. final def getDefault[T](param: Param[T]): Option[T]
    Definition Classes
    Params
  64. def getDoSample: Boolean

    Definition Classes
    GPTGenerationParams
  65. def getEngine: String
    Definition Classes
    HasEngine
  66. def getIgnoreTokenIds: Array[Int]

    Definition Classes
    GPTGenerationParams
  67. def getInputCols: Array[String]
    Definition Classes
    HasInputAnnotationCols
  68. def getLazyAnnotator: Boolean
    Definition Classes
    CanBeLazy
  69. def getMaxContextLength: Int

    Definition Classes
    GPTGenerationParams
  70. def getMaxNewTokens: Int

    Definition Classes
    GPTGenerationParams
  71. def getMaxTextLength: Int

    Definition Classes
    MedicalTextGenerator
  72. def getMlFrameworkType: String

    Get ML framework type

    Get ML framework type

    Definition Classes
    MedicalTextGenerator
  73. def getModelIfNotSet: MedicalEncoderDecoderModel

    Definition Classes
    MedicalTextGenerator
  74. def getModelType: String

    Get model type

    Get model type

    Definition Classes
    MedicalTextGenerator
  75. def getNoRepeatNgramSize: Int

    Definition Classes
    GPTGenerationParams
  76. final def getOrDefault[T](param: Param[T]): T
    Definition Classes
    Params
  77. final def getOutputCol: String
    Definition Classes
    HasOutputAnnotationCol
  78. def getParam(paramName: String): Param[Any]
    Definition Classes
    Params
  79. def getPostProcessingSubstitutionsStr: String
  80. def getRandomSeed: Option[Int]

    Definition Classes
    GPTGenerationParams
  81. def getSQLSchemaStr: String
  82. def getSignatures: Option[Map[String, String]]
    Definition Classes
    MedicalTextGenerator
  83. def getStopAtEos: Boolean

    Checks whether text generation stops when the end-of-sentence token is encountered.

    Checks whether text generation stops when the end-of-sentence token is encountered.

    Definition Classes
    MedicalTextGenerator
  84. def getTableLimit: Int

    Get the maximum number of tables allowed

  85. def getTopK: Int

    Definition Classes
    GPTGenerationParams
  86. def getUseCache: Boolean
    Definition Classes
    MedicalTextGenerator
  87. final def hasDefault[T](param: Param[T]): Boolean
    Definition Classes
    Params
  88. def hasParam(paramName: String): Boolean
    Definition Classes
    Params
  89. def hasParent: Boolean
    Definition Classes
    Model
  90. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  91. val ignoreTokenIds: IntArrayParam

    A list of token ids which are ignored in the decoder's output (Default: Array())

    A list of token ids which are ignored in the decoder's output (Default: Array())

    Definition Classes
    GPTGenerationParams
  92. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  93. def initializeLogIfNecessary(isInterpreter: Boolean): Unit
    Attributes
    protected
    Definition Classes
    Logging
  94. val inputAnnotatorTypes: Array[String]

    Input annotator type : DOCUMENT, DOCUMENT

    Input annotator type : DOCUMENT, DOCUMENT

    Definition Classes
    MedicalTextGenerator → HasInputAnnotationCols
  95. final val inputCols: StringArrayParam
    Attributes
    protected
    Definition Classes
    HasInputAnnotationCols
  96. final def isDefined(param: Param[_]): Boolean
    Definition Classes
    Params
  97. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  98. final def isSet(param: Param[_]): Boolean
    Definition Classes
    Params
  99. def isTraceEnabled(): Boolean
    Attributes
    protected
    Definition Classes
    Logging
  100. val lazyAnnotator: BooleanParam
    Definition Classes
    CanBeLazy
  101. def log: Logger
    Attributes
    protected
    Definition Classes
    Logging
  102. def logDebug(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  103. def logDebug(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  104. def logError(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  105. def logError(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  106. def logInfo(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  107. def logInfo(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  108. def logName: String
    Attributes
    protected
    Definition Classes
    Logging
  109. def logTrace(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  110. def logTrace(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  111. def logWarning(msg: ⇒ String, throwable: Throwable): Unit
    Attributes
    protected
    Definition Classes
    Logging
  112. def logWarning(msg: ⇒ String): Unit
    Attributes
    protected
    Definition Classes
    Logging
  113. val maxContextLength: IntParam

    Maximum length of context text.

    Maximum length of context text. (Default: 1024)

    Definition Classes
    GPTGenerationParams
  114. val maxNewTokens: IntParam

    Maximum number of new tokens to be generated (Default: 30)

    Maximum number of new tokens to be generated (Default: 30)

    Definition Classes
    GPTGenerationParams
  115. val maxTextLength: IntParam

    Maximum length of context text.

    Maximum length of context text. (Default: 1024)

    Definition Classes
    MedicalTextGenerator
  116. val merges: MapFeature[(String, String), Int]

    Holding merges.txt coming from RoBERTa model

    Holding merges.txt coming from RoBERTa model

    Definition Classes
    MedicalTextGenerator
  117. val mlFrameworkType: Param[String]

    ML framework type

    ML framework type

    Definition Classes
    MedicalTextGenerator
  118. val modelType: Param[String]

    Model type

    Model type

    Definition Classes
    MedicalTextGenerator
  119. def msgHelper(schema: StructType): String
    Attributes
    protected
    Definition Classes
    HasInputAnnotationCols
  120. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  121. val noRepeatNgramSize: IntParam

    If set to int > 0, all ngrams of that size can only occur once (Default: 0)

    If set to int > 0, all ngrams of that size can only occur once (Default: 0)

    Definition Classes
    GPTGenerationParams
  122. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  123. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  124. def onWrite(path: String, spark: SparkSession): Unit
    Definition Classes
    MedicalTextGenerator → ParamsAndFeaturesWritable
  125. val optionalInputAnnotatorTypes: Array[String]
    Definition Classes
    HasInputAnnotationCols
  126. val outputAnnotatorType: String

    Output annotator type : DOCUMENT

    Output annotator type : DOCUMENT

    Definition Classes
    MedicalTextGenerator → HasOutputAnnotatorType
  127. final val outputCol: Param[String]
    Attributes
    protected
    Definition Classes
    HasOutputAnnotationCol
  128. lazy val params: Array[Param[_]]
    Definition Classes
    Params
  129. var parent: Estimator[MedicalQuestionAnswering]
    Definition Classes
    Model
  130. def postProcessing(text: String): String
  131. var postProcessingSubstitutions: MapFeature[String, String]

    List of strings and their substitution to be applied in post-processing

  132. val randomSeed: Option[Int]

    Optional Random seed for the model.

    Optional Random seed for the model. Needs to be of type Long.

    Definition Classes
    GPTGenerationParams
  133. def save(path: String): Unit
    Definition Classes
    MLWritable
    Annotations
    @Since( "1.6.0" ) @throws( ... )
  134. var schema: MapFeature[String, Array[String]]

    SQL schema

  135. def set[T](feature: StructFeature[T], value: T): Text2SQL.this.type
    Attributes
    protected
    Definition Classes
    HasFeatures
  136. def set[K, V](feature: MapFeature[K, V], value: Map[K, V]): Text2SQL.this.type
    Attributes
    protected
    Definition Classes
    HasFeatures
  137. def set[T](feature: SetFeature[T], value: Set[T]): Text2SQL.this.type
    Attributes
    protected
    Definition Classes
    HasFeatures
  138. def set[T](feature: ArrayFeature[T], value: Array[T]): Text2SQL.this.type
    Attributes
    protected
    Definition Classes
    HasFeatures
  139. final def set(paramPair: ParamPair[_]): Text2SQL.this.type
    Attributes
    protected
    Definition Classes
    Params
  140. final def set(param: String, value: Any): Text2SQL.this.type
    Attributes
    protected
    Definition Classes
    Params
  141. final def set[T](param: Param[T], value: T): Text2SQL.this.type
    Definition Classes
    Params
  142. def setAdditionalTokens(values: HashMap[Int, String]): Text2SQL.this.type

    Set additional tokens

    Set additional tokens

    Definition Classes
    MedicalTextGenerator
  143. def setAdditionalTokens(value: Map[Int, String]): Text2SQL.this.type

    Set additional tokens

    Set additional tokens

    Definition Classes
    MedicalTextGenerator
  144. def setBatchSize(size: Int): Text2SQL.this.type
    Definition Classes
    HasBatchedAnnotate
  145. def setCaseSensitive(value: Boolean): Text2SQL.this.type
    Definition Classes
    HasCaseSensitiveProperties
  146. def setConfigProtoBytes(bytes: Array[Int]): Text2SQL.this.type

    Definition Classes
    MedicalTextGenerator
  147. def setCustomPrompt(value: String): Text2SQL.this.type

    Set custom model prompt

    Set custom model prompt

    Definition Classes
    MedicalTextGenerator
  148. def setDefault[T](feature: StructFeature[T], value: () ⇒ T): Text2SQL.this.type
    Attributes
    protected
    Definition Classes
    HasFeatures
  149. def setDefault[K, V](feature: MapFeature[K, V], value: () ⇒ Map[K, V]): Text2SQL.this.type
    Attributes
    protected
    Definition Classes
    HasFeatures
  150. def setDefault[T](feature: SetFeature[T], value: () ⇒ Set[T]): Text2SQL.this.type
    Attributes
    protected
    Definition Classes
    HasFeatures
  151. def setDefault[T](feature: ArrayFeature[T], value: () ⇒ Array[T]): Text2SQL.this.type
    Attributes
    protected
    Definition Classes
    HasFeatures
  152. final def setDefault(paramPairs: ParamPair[_]*): Text2SQL.this.type
    Attributes
    protected
    Definition Classes
    Params
  153. final def setDefault[T](param: Param[T], value: T): Text2SQL.this.type
    Attributes
    protected
    Definition Classes
    Params
  154. def setDoSample(value: Boolean): Text2SQL.this.type

    Definition Classes
    GPTGenerationParams
  155. def setIgnoreTokenIds(tokenIds: Array[Int]): Text2SQL.this.type

    Definition Classes
    GPTGenerationParams
  156. final def setInputCols(value: String*): Text2SQL.this.type
    Definition Classes
    HasInputAnnotationCols
  157. def setInputCols(value: Array[String]): Text2SQL.this.type
    Definition Classes
    HasInputAnnotationCols
  158. def setLazyAnnotator(value: Boolean): Text2SQL.this.type
    Definition Classes
    CanBeLazy
  159. def setMaxContextLength(value: Int): Text2SQL.this.type

    Definition Classes
    GPTGenerationParams
  160. def setMaxNewTokens(value: Int): Text2SQL.this.type

    Definition Classes
    GPTGenerationParams
  161. def setMaxTextLength(value: Int): Text2SQL.this.type

    Definition Classes
    MedicalTextGenerator
  162. def setMerges(value: Map[(String, String), Int]): Text2SQL.this.type

    Definition Classes
    MedicalTextGenerator
  163. def setMlFrameworkType(value: String): Text2SQL.this.type

    Set ML framework type

    Set ML framework type

    Definition Classes
    MedicalTextGenerator
  164. def setModelIfNotSet(spark: SparkSession, model: MedicalEncoderDecoderModel): Text2SQL.this.type
    Definition Classes
    MedicalTextGenerator
  165. def setModelIfNotSet(spark: SparkSession, encoder: OnnxWrapper, decoder: OnnxWrapper, spp: SentencePieceWrapper): Text2SQL.this.type
    Definition Classes
    MedicalTextGenerator
  166. def setModelIfNotSet(spark: SparkSession, tfWrapper: TensorflowWrapper, spp: SentencePieceWrapper, useCache: Boolean): Text2SQL.this.type
    Definition Classes
    MedicalTextGenerator
  167. def setModelIfNotSet(spark: SparkSession, tfWrapper: TensorflowWrapper): Text2SQL.this.type

    Definition Classes
    MedicalTextGenerator
  168. def setModelType(value: String): Text2SQL.this.type

    Set model type

    Set model type

    Definition Classes
    MedicalTextGenerator
  169. def setNoRepeatNgramSize(value: Int): Text2SQL.this.type

    Definition Classes
    GPTGenerationParams
  170. final def setOutputCol(value: String): Text2SQL.this.type
    Definition Classes
    HasOutputAnnotationCol
  171. def setParent(parent: Estimator[MedicalQuestionAnswering]): MedicalQuestionAnswering
    Definition Classes
    Model
  172. def setPostProcessingSubstitutions(substitutions: HashMap[String, String]): Text2SQL.this.type

    Set post processing substitution strings

  173. def setPostProcessingSubstitutions(substitutions: Map[String, String]): Text2SQL.this.type

    Set post processing substitution strings

  174. def setRandomSeed(value: Int): Text2SQL.this.type

    Definition Classes
    GPTGenerationParams
  175. def setSchema(schema: HashMap[String, List[String]]): Text2SQL.this.type

    Set SQL schema

  176. def setSchema(schema: Map[String, Array[String]]): Text2SQL.this.type

    Set SQL schema

  177. def setSignatures(value: Map[String, String]): Text2SQL.this.type

    Definition Classes
    MedicalTextGenerator
  178. def setStopAtEos(value: Boolean): Text2SQL.this.type

    Determines whether text generation stops when the end-of-sentence token is encountered.

    Determines whether text generation stops when the end-of-sentence token is encountered.

    Definition Classes
    MedicalTextGenerator
  179. def setTableLimit(value: Int): Text2SQL.this.type

    Set the maximum number of tables allowed

  180. def setTopK(value: Int): Text2SQL.this.type

    Definition Classes
    GPTGenerationParams
  181. def setUseCache(value: Boolean): Text2SQL.this.type
    Definition Classes
    MedicalTextGenerator
  182. def setVocabulary(value: Map[String, Int]): Text2SQL.this.type

    Definition Classes
    MedicalTextGenerator
  183. val signatures: MapFeature[String, String]

    It contains TF model signatures for the laded saved model

    It contains TF model signatures for the laded saved model

    Definition Classes
    MedicalTextGenerator
  184. val stopAtEos: BooleanParam

    Stop text generation when the end-of-sentence token is encountered.

    Stop text generation when the end-of-sentence token is encountered.

    Definition Classes
    MedicalTextGenerator
  185. final def synchronized[T0](arg0: ⇒ T0): T0
    Definition Classes
    AnyRef
  186. var tableLimit: IntParam

    Maximum number of tables allowed

  187. def toString(): String
    Definition Classes
    Identifiable → AnyRef → Any
  188. val topK: IntParam

    The number of highest probability vocabulary tokens to consider

    The number of highest probability vocabulary tokens to consider

    Definition Classes
    GPTGenerationParams
  189. final def transform(dataset: Dataset[_]): DataFrame
    Definition Classes
    AnnotatorModel → Transformer
  190. def transform(dataset: Dataset[_], paramMap: ParamMap): DataFrame
    Definition Classes
    Transformer
    Annotations
    @Since( "2.0.0" )
  191. def transform(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): DataFrame
    Definition Classes
    Transformer
    Annotations
    @Since( "2.0.0" ) @varargs()
  192. final def transformSchema(schema: StructType): StructType
    Definition Classes
    RawAnnotator → PipelineStage
  193. def transformSchema(schema: StructType, logging: Boolean): StructType
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()
  194. val uid: String
    Definition Classes
    Text2SQLMedicalTextGenerator → Identifiable
  195. val useCache: BooleanParam

    Cache internal state of the model to improve performance

    Cache internal state of the model to improve performance

    Definition Classes
    MedicalTextGenerator
  196. def validate(schema: StructType): Boolean
    Attributes
    protected
    Definition Classes
    RawAnnotator
  197. val vocabulary: MapFeature[String, Int]

    Vocabulary used to encode the words to ids with bpeTokenizer.encode

    Vocabulary used to encode the words to ids with bpeTokenizer.encode

    Definition Classes
    MedicalTextGenerator
  198. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  199. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  200. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws( ... ) @native()
  201. def wrapColumnMetadata(col: Column): Column
    Attributes
    protected
    Definition Classes
    RawAnnotator
  202. def write: MLWriter
    Definition Classes
    ParamsAndFeaturesWritable → DefaultParamsWritable → MLWritable
  203. def writeOnnxModel(path: String, spark: SparkSession, onnxWrapper: OnnxWrapper, suffix: String, fileName: String): Unit
    Definition Classes
    WriteOnnxModel
  204. def writeOnnxModels(path: String, spark: SparkSession, onnxWrappersWithNames: Seq[(OnnxWrapper, String)], suffix: String, dataFileSuffix: String): Unit
    Definition Classes
    WriteOnnxModel
  205. def writeSentencePieceModel(path: String, spark: SparkSession, spp: SentencePieceWrapper, suffix: String, filename: String): Unit
    Definition Classes
    WriteSentencePieceModel
  206. def writeTensorflowHub(path: String, tfPath: String, spark: SparkSession, suffix: String): Unit
    Definition Classes
    WriteTensorflowModel
  207. def writeTensorflowModel(path: String, spark: SparkSession, tensorflow: TensorflowWrapper, suffix: String, filename: String, configProtoBytes: Option[Array[Byte]]): Unit
    Definition Classes
    WriteTensorflowModel
  208. def writeTensorflowModelV2(path: String, spark: SparkSession, tensorflow: TensorflowWrapper, suffix: String, filename: String, configProtoBytes: Option[Array[Byte]], savedSignatures: Option[Map[String, String]]): Unit
    Definition Classes
    WriteTensorflowModel

Inherited from MedicalTextGenerator

Inherited from CheckLicense

Inherited from HasEngine

Inherited from WriteSentencePieceModel

Inherited from WriteOnnxModel

Inherited from WriteTensorflowModel

Inherited from HasBatchedAnnotate[MedicalQuestionAnswering]

Inherited from HasCaseSensitiveProperties

Inherited from GPTGenerationParams

Inherited from AnnotatorModel[MedicalQuestionAnswering]

Inherited from CanBeLazy

Inherited from RawAnnotator[MedicalQuestionAnswering]

Inherited from HasOutputAnnotationCol

Inherited from HasInputAnnotationCols

Inherited from HasOutputAnnotatorType

Inherited from ParamsAndFeaturesWritable

Inherited from HasFeatures

Inherited from DefaultParamsWritable

Inherited from MLWritable

Inherited from Model[MedicalQuestionAnswering]

Inherited from Transformer

Inherited from PipelineStage

Inherited from Logging

Inherited from Params

Inherited from Serializable

Inherited from Serializable

Inherited from Identifiable

Inherited from AnyRef

Inherited from Any

getParam

param

setParam

Ungrouped