org.apache.spark.mllib.feature

Class Word2Vec

Object

org.apache.spark.mllib.feature.Word2Vec

All Implemented Interfaces:

java.io.Serializable, Logging

public class Word2Vec
extends Object
implements scala.Serializable, Logging

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
Word2Vec()

Method Summary

Methods

Modifier and Type	Method and Description
<S extends Iterable<String>> Word2VecModel	fit(JavaRDD<S> dataset)
<S extends scala.collection.Iterable<String>> Word2VecModel	fit(RDD<S> dataset)
Word2Vec	setLearningRate(double learningRate)
Word2Vec	setMinCount(int minCount)
Word2Vec	setNumIterations(int numIterations)
Word2Vec	setNumPartitions(int numPartitions)
Word2Vec	setSeed(long seed)
Word2Vec	setVectorSize(int vectorSize)
Word2Vec	setWindowSize(int window)

Methods inherited from class Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.apache.spark.Logging

initializeIfNecessary, initializeLogging, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning

Constructor Detail

Word2Vec

public Word2Vec()

Method Detail

setVectorSize

public Word2Vec setVectorSize(int vectorSize)

setLearningRate

public Word2Vec setLearningRate(double learningRate)

setNumPartitions

public Word2Vec setNumPartitions(int numPartitions)

setNumIterations

public Word2Vec setNumIterations(int numIterations)

setSeed

public Word2Vec setSeed(long seed)

setWindowSize

public Word2Vec setWindowSize(int window)

setMinCount

public Word2Vec setMinCount(int minCount)

fit

public <S extends scala.collection.Iterable<String>> Word2VecModel fit(RDD<S> dataset)

fit

public <S extends Iterable<String>> Word2VecModel fit(JavaRDD<S> dataset)