Class/Object

com.johnsnowlabs.nlp.embeddings

WordEmbeddings

Related Docs: object WordEmbeddings | package embeddings

Permalink

class WordEmbeddings extends AnnotatorApproach[WordEmbeddingsModel] with HasWordEmbeddings

Linear Supertypes
HasWordEmbeddings, HasEmbeddings, AnnotatorApproach[WordEmbeddingsModel], DefaultParamsWritable, MLWritable, HasOutputAnnotatorType, HasOutputAnnotationCol, HasInputAnnotationCols, Estimator[WordEmbeddingsModel], PipelineStage, Logging, Params, Serializable, Serializable, Identifiable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. WordEmbeddings
  2. HasWordEmbeddings
  3. HasEmbeddings
  4. AnnotatorApproach
  5. DefaultParamsWritable
  6. MLWritable
  7. HasOutputAnnotatorType
  8. HasOutputAnnotationCol
  9. HasInputAnnotationCols
  10. Estimator
  11. PipelineStage
  12. Logging
  13. Params
  14. Serializable
  15. Serializable
  16. Identifiable
  17. AnyRef
  18. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new WordEmbeddings()

    Permalink
  2. new WordEmbeddings(uid: String)

    Permalink

Type Members

  1. type AnnotatorType = String

    Permalink
    Definition Classes
    HasOutputAnnotatorType

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 $[T](param: Param[T]): T

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  4. final def ==(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  5. final def asInstanceOf[T0]: T0

    Permalink
    Definition Classes
    Any
  6. def beforeTraining(sparkSession: SparkSession): Unit

    Permalink
    Definition Classes
    WordEmbeddingsAnnotatorApproach
  7. val caseSensitive: BooleanParam

    Permalink
    Definition Classes
    HasEmbeddings
  8. final def checkSchema(schema: StructType, inputAnnotatorType: String): Boolean

    Permalink
    Attributes
    protected
    Definition Classes
    HasInputAnnotationCols
  9. final def clear(param: Param[_]): WordEmbeddings.this.type

    Permalink
    Definition Classes
    Params
  10. def clone(): AnyRef

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  11. final def copy(extra: ParamMap): Estimator[WordEmbeddingsModel]

    Permalink
    Definition Classes
    AnnotatorApproach → Estimator → PipelineStage → Params
  12. def copyValues[T <: Params](to: T, extra: ParamMap): T

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  13. final def defaultCopy[T <: Params](extra: ParamMap): T

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  14. val description: String

    Permalink
    Definition Classes
    WordEmbeddingsAnnotatorApproach
  15. val dimension: IntParam

    Permalink
    Definition Classes
    HasEmbeddings
  16. val embeddingsFormat: IntParam

    Permalink
  17. val embeddingsRef: Param[String]

    Permalink
    Definition Classes
    HasWordEmbeddings
  18. final def eq(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  19. def equals(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  20. def explainParam(param: Param[_]): String

    Permalink
    Definition Classes
    Params
  21. def explainParams(): String

    Permalink
    Definition Classes
    Params
  22. final def extractParamMap(): ParamMap

    Permalink
    Definition Classes
    Params
  23. final def extractParamMap(extra: ParamMap): ParamMap

    Permalink
    Definition Classes
    Params
  24. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  25. final def fit(dataset: Dataset[_]): WordEmbeddingsModel

    Permalink
    Definition Classes
    AnnotatorApproach → Estimator
  26. def fit(dataset: Dataset[_], paramMaps: Array[ParamMap]): Seq[WordEmbeddingsModel]

    Permalink
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  27. def fit(dataset: Dataset[_], paramMap: ParamMap): WordEmbeddingsModel

    Permalink
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" )
  28. def fit(dataset: Dataset[_], firstParamPair: ParamPair[_], otherParamPairs: ParamPair[_]*): WordEmbeddingsModel

    Permalink
    Definition Classes
    Estimator
    Annotations
    @Since( "2.0.0" ) @varargs()
  29. final def get[T](param: Param[T]): Option[T]

    Permalink
    Definition Classes
    Params
  30. def getCaseSensitive: Boolean

    Permalink
    Definition Classes
    HasEmbeddings
  31. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any
  32. def getClusterEmbeddings: ClusterWordEmbeddings

    Permalink
    Attributes
    protected
    Definition Classes
    HasWordEmbeddings
  33. final def getDefault[T](param: Param[T]): Option[T]

    Permalink
    Definition Classes
    Params
  34. def getDimension: Int

    Permalink
    Definition Classes
    HasEmbeddings
  35. def getEmbeddings: WordEmbeddingsRetriever

    Permalink
    Attributes
    protected
    Definition Classes
    HasWordEmbeddings
  36. def getEmbeddingsFormat: String

    Permalink
  37. def getEmbeddingsRef: String

    Permalink
    Definition Classes
    HasWordEmbeddings
  38. def getIncludeEmbeddings: Boolean

    Permalink
    Definition Classes
    HasWordEmbeddings
  39. def getInputCols: Array[String]

    Permalink

    returns

    input annotations columns currently used

    Definition Classes
    HasInputAnnotationCols
  40. final def getOrDefault[T](param: Param[T]): T

    Permalink
    Definition Classes
    Params
  41. final def getOutputCol: String

    Permalink

    Gets annotation column name going to generate

    Gets annotation column name going to generate

    Definition Classes
    HasOutputAnnotationCol
  42. def getParam(paramName: String): Param[Any]

    Permalink
    Definition Classes
    Params
  43. def getSourcePath: String

    Permalink
  44. final def hasDefault[T](param: Param[T]): Boolean

    Permalink
    Definition Classes
    Params
  45. def hasParam(paramName: String): Boolean

    Permalink
    Definition Classes
    Params
  46. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  47. val includeEmbeddings: BooleanParam

    Permalink
    Definition Classes
    HasWordEmbeddings
  48. def initializeLogIfNecessary(isInterpreter: Boolean, silent: Boolean): Boolean

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  49. def initializeLogIfNecessary(isInterpreter: Boolean): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  50. val inputAnnotatorTypes: Array[String]

    Permalink

    Annotator reference id.

    Annotator reference id. Used to identify elements in metadata or to refer to this annotator type

    Definition Classes
    WordEmbeddingsHasInputAnnotationCols
  51. final val inputCols: StringArrayParam

    Permalink

    columns that contain annotations necessary to run this annotator AnnotatorType is used both as input and output columns if not specified

    columns that contain annotations necessary to run this annotator AnnotatorType is used both as input and output columns if not specified

    Attributes
    protected
    Definition Classes
    HasInputAnnotationCols
  52. final def isDefined(param: Param[_]): Boolean

    Permalink
    Definition Classes
    Params
  53. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any
  54. def isLoaded(): Boolean

    Permalink
    Attributes
    protected
    Definition Classes
    HasWordEmbeddings
  55. final def isSet(param: Param[_]): Boolean

    Permalink
    Definition Classes
    Params
  56. def isTraceEnabled(): Boolean

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  57. def log: Logger

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  58. def logDebug(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  59. def logDebug(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  60. def logError(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  61. def logError(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  62. def logInfo(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  63. def logInfo(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  64. def logName: String

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  65. def logTrace(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  66. def logTrace(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  67. def logWarning(msg: ⇒ String, throwable: Throwable): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  68. def logWarning(msg: ⇒ String): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    Logging
  69. final def ne(arg0: AnyRef): Boolean

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

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

    Permalink
    Definition Classes
    AnyRef
  72. def onTrained(model: WordEmbeddingsModel, spark: SparkSession): Unit

    Permalink
    Definition Classes
    AnnotatorApproach
  73. val outputAnnotatorType: AnnotatorType

    Permalink
    Definition Classes
    WordEmbeddingsHasOutputAnnotatorType
  74. final val outputCol: Param[String]

    Permalink
    Attributes
    protected
    Definition Classes
    HasOutputAnnotationCol
  75. lazy val params: Array[Param[_]]

    Permalink
    Definition Classes
    Params
  76. var preloadedEmbeddings: Option[ClusterWordEmbeddings]

    Permalink
    Attributes
    protected
    Definition Classes
    HasWordEmbeddings
  77. def save(path: String): Unit

    Permalink
    Definition Classes
    MLWritable
    Annotations
    @Since( "1.6.0" ) @throws( ... )
  78. final def set(paramPair: ParamPair[_]): WordEmbeddings.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  79. final def set(param: String, value: Any): WordEmbeddings.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  80. final def set[T](param: Param[T], value: T): WordEmbeddings.this.type

    Permalink
    Definition Classes
    Params
  81. def setAsLoaded(): Unit

    Permalink
    Attributes
    protected
    Definition Classes
    HasWordEmbeddings
  82. def setCaseSensitive(value: Boolean): WordEmbeddings.this.type

    Permalink
    Definition Classes
    HasEmbeddings
  83. final def setDefault(paramPairs: ParamPair[_]*): WordEmbeddings.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  84. final def setDefault[T](param: Param[T], value: T): WordEmbeddings.this.type

    Permalink
    Attributes
    protected
    Definition Classes
    Params
  85. def setDimension(value: Int): WordEmbeddings.this.type

    Permalink
    Definition Classes
    HasEmbeddings
  86. def setEmbeddingsFormat(format: String): WordEmbeddings.this.type

    Permalink
  87. def setEmbeddingsRef(value: String): WordEmbeddings.this.type

    Permalink
    Definition Classes
    HasWordEmbeddings
  88. def setEmbeddingsSource(path: String, nDims: Int, format: String): WordEmbeddings.this.type

    Permalink
  89. def setEmbeddingsSource(path: String, nDims: Int, format: Format): WordEmbeddings.this.type

    Permalink
  90. def setIncludeEmbeddings(value: Boolean): WordEmbeddings.this.type

    Permalink
    Definition Classes
    HasWordEmbeddings
  91. final def setInputCols(value: String*): WordEmbeddings.this.type

    Permalink
    Definition Classes
    HasInputAnnotationCols
  92. final def setInputCols(value: Array[String]): WordEmbeddings.this.type

    Permalink

    Overrides required annotators column if different than default

    Overrides required annotators column if different than default

    Definition Classes
    HasInputAnnotationCols
  93. final def setOutputCol(value: String): WordEmbeddings.this.type

    Permalink

    Overrides annotation column name when transforming

    Overrides annotation column name when transforming

    Definition Classes
    HasOutputAnnotationCol
  94. def setSourcePath(path: String): WordEmbeddings.this.type

    Permalink
  95. val sourceEmbeddingsPath: Param[String]

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

    Permalink
    Definition Classes
    AnyRef
  97. def toString(): String

    Permalink
    Definition Classes
    Identifiable → AnyRef → Any
  98. def train(dataset: Dataset[_], recursivePipeline: Option[PipelineModel]): WordEmbeddingsModel

    Permalink
    Definition Classes
    WordEmbeddingsAnnotatorApproach
  99. final def transformSchema(schema: StructType): StructType

    Permalink

    requirement for pipeline transformation validation.

    requirement for pipeline transformation validation. It is called on fit()

    Definition Classes
    AnnotatorApproach → PipelineStage
  100. def transformSchema(schema: StructType, logging: Boolean): StructType

    Permalink
    Attributes
    protected
    Definition Classes
    PipelineStage
    Annotations
    @DeveloperApi()
  101. val uid: String

    Permalink
    Definition Classes
    WordEmbeddings → Identifiable
  102. def validate(schema: StructType): Boolean

    Permalink

    takes a Dataset and checks to see if all the required annotation types are present.

    takes a Dataset and checks to see if all the required annotation types are present.

    schema

    to be validated

    returns

    True if all the required types are present, else false

    Attributes
    protected
    Definition Classes
    AnnotatorApproach
  103. final def wait(): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  104. final def wait(arg0: Long, arg1: Int): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  105. final def wait(arg0: Long): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  106. def wrapEmbeddingsMetadata(col: Column, embeddingsDim: Int, embeddingsRef: Option[String] = None): Column

    Permalink
    Attributes
    protected
    Definition Classes
    HasEmbeddings
  107. def write: MLWriter

    Permalink
    Definition Classes
    DefaultParamsWritable → MLWritable

Inherited from HasWordEmbeddings

Inherited from HasEmbeddings

Inherited from DefaultParamsWritable

Inherited from MLWritable

Inherited from HasOutputAnnotatorType

Inherited from HasOutputAnnotationCol

Inherited from HasInputAnnotationCols

Inherited from Estimator[WordEmbeddingsModel]

Inherited from PipelineStage

Inherited from Logging

Inherited from Params

Inherited from Serializable

Inherited from Serializable

Inherited from Identifiable

Inherited from AnyRef

Inherited from Any

Ungrouped