Packages

class Word2VecModel extends Serializable with Saveable

Word2Vec model

Annotations
@Since("1.1.0")
Linear Supertypes
Saveable, Serializable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. Word2VecModel
  2. Saveable
  3. Serializable
  4. AnyRef
  5. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. Protected

Instance Constructors

  1. new Word2VecModel(model: Map[String, Array[Float]])
    Annotations
    @Since("1.5.0")

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. final def asInstanceOf[T0]: T0
    Definition Classes
    Any
  5. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.CloneNotSupportedException]) @native()
  6. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  7. def equals(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef → Any
  8. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.Throwable])
  9. def findSynonyms(vector: Vector, num: Int): Array[(String, Double)]

    Find synonyms of the vector representation of a word, possibly including any words in the model vocabulary whose vector representation is the supplied vector.

    Find synonyms of the vector representation of a word, possibly including any words in the model vocabulary whose vector representation is the supplied vector.

    vector

    vector representation of a word

    num

    number of synonyms to find

    returns

    array of (word, cosineSimilarity)

    Annotations
    @Since("1.1.0")
  10. def findSynonyms(word: String, num: Int): Array[(String, Double)]

    Find synonyms of a word; do not include the word itself in results.

    Find synonyms of a word; do not include the word itself in results.

    word

    a word

    num

    number of synonyms to find

    returns

    array of (word, cosineSimilarity)

    Annotations
    @Since("1.1.0")
  11. final def getClass(): Class[_ <: AnyRef]
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  12. def getVectors: Map[String, Array[Float]]

    Returns a map of words to their vector representations.

    Returns a map of words to their vector representations.

    Annotations
    @Since("1.2.0")
  13. def hashCode(): Int
    Definition Classes
    AnyRef → Any
    Annotations
    @native()
  14. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  15. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  16. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  17. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native()
  18. def save(sc: SparkContext, path: String): Unit

    Save this model to the given path.

    Save this model to the given path.

    This saves:

    • human-readable (JSON) model metadata to path/metadata/
    • Parquet formatted data to path/data/

    The model may be loaded using Loader.load.

    sc

    Spark context used to save model data.

    path

    Path specifying the directory in which to save this model. If the directory already exists, this method throws an exception.

    Definition Classes
    Word2VecModelSaveable
    Annotations
    @Since("1.4.0")
  19. final def synchronized[T0](arg0: => T0): T0
    Definition Classes
    AnyRef
  20. def toString(): String
    Definition Classes
    AnyRef → Any
  21. def transform(word: String): Vector

    Transforms a word to its vector representation

    Transforms a word to its vector representation

    word

    a word

    returns

    vector representation of word

    Annotations
    @Since("1.1.0")
  22. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.InterruptedException])
  23. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.InterruptedException])
  24. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.InterruptedException]) @native()

Inherited from Saveable

Inherited from Serializable

Inherited from AnyRef

Inherited from Any

Ungrouped