Class/Object

org.apache.spark.mllib.feature

Word2VecModel

Related Docs: object Word2VecModel | package feature

Permalink

class Word2VecModel extends Serializable with Saveable

Word2Vec model

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

Instance Constructors

  1. new Word2VecModel(model: Map[String, Array[Float]])

    Permalink
    Annotations
    @Since( "1.5.0" )

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. final def eq(arg0: AnyRef): Boolean

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

    Permalink
    Definition Classes
    AnyRef → Any
  8. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  9. def findSynonyms(vector: Vector, num: Int): Array[(String, Double)]

    Permalink

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

    Find synonyms of the vector representation of a word, possibly including any words in the model vocabulary whose vector respresentation 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)]

    Permalink

    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. def formatVersion: String

    Permalink

    Current version of model save/load format.

    Current version of model save/load format.

    Attributes
    protected
    Definition Classes
    Word2VecModelSaveable
  12. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any
  13. def getVectors: Map[String, Array[Float]]

    Permalink

    Returns a map of words to their vector representations.

    Returns a map of words to their vector representations.

    Annotations
    @Since( "1.2.0" )
  14. def hashCode(): Int

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

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

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

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

    Permalink
    Definition Classes
    AnyRef
  19. def save(sc: SparkContext, path: String): Unit

    Permalink

    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" )
  20. final def synchronized[T0](arg0: ⇒ T0): T0

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

    Permalink
    Definition Classes
    AnyRef → Any
  22. def transform(word: String): Vector

    Permalink

    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" )
  23. final def wait(): Unit

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

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

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

Inherited from Saveable

Inherited from Serializable

Inherited from Serializable

Inherited from AnyRef

Inherited from Any

Ungrouped