class Word2VecModel extends Serializable with Saveable
Word2Vec model
- Annotations
- @Since( "1.1.0" )
- Alphabetic
- By Inheritance
- Word2VecModel
- Saveable
- Serializable
- Serializable
- AnyRef
- Any
- Hide All
- Show All
- Public
- All
Instance Constructors
-
new
Word2VecModel(model: Map[String, Array[Float]])
- Annotations
- @Since( "1.5.0" )
Value Members
-
final
def
!=(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
##(): Int
- Definition Classes
- AnyRef → Any
-
final
def
==(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
asInstanceOf[T0]: T0
- Definition Classes
- Any
-
def
clone(): AnyRef
- Attributes
- protected[java.lang]
- Definition Classes
- AnyRef
- Annotations
- @native() @throws( ... )
-
final
def
eq(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
def
equals(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
def
finalize(): Unit
- Attributes
- protected[java.lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( classOf[java.lang.Throwable] )
-
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 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" )
-
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" )
-
def
formatVersion: String
Current version of model save/load format.
Current version of model save/load format.
- Attributes
- protected
- Definition Classes
- Word2VecModel → Saveable
-
final
def
getClass(): Class[_]
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
-
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" )
-
def
hashCode(): Int
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
-
final
def
isInstanceOf[T0]: Boolean
- Definition Classes
- Any
-
final
def
ne(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
final
def
notify(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
-
final
def
notifyAll(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
-
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
- Word2VecModel → Saveable
- Annotations
- @Since( "1.4.0" )
-
final
def
synchronized[T0](arg0: ⇒ T0): T0
- Definition Classes
- AnyRef
-
def
toString(): String
- Definition Classes
- AnyRef → Any
-
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" )
-
final
def
wait(): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long, arg1: Int): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long): Unit
- Definition Classes
- AnyRef
- Annotations
- @native() @throws( ... )