org.deeplearning4j.spark.models.embeddings.word2vec

Class Word2VecParam.Builder

java.lang.Object

org.deeplearning4j.spark.models.embeddings.word2vec.Word2VecParam.Builder

Enclosing class:

Word2VecParam

public static class Word2VecParam.Builder
extends java.lang.Object

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
Word2VecParam	build()
Word2VecParam.Builder	expTable(org.apache.spark.broadcast.Broadcast<double[]> expTable)
Word2VecParam.Builder	lastChecked(int lastChecked)
Word2VecParam.Builder	negative(double negative)
Word2VecParam.Builder	numWords(int numWords)
Word2VecParam.Builder	setAlpha(double alpha)
Word2VecParam.Builder	setMinAlpha(double minAlpha)
Word2VecParam.Builder	setNextRandom(java.util.concurrent.atomic.AtomicLong nextRandom)
Word2VecParam.Builder	setVectorLength(int vectorLength)
Word2VecParam.Builder	table(org.nd4j.linalg.api.ndarray.INDArray table)
Word2VecParam.Builder	totalWords(int totalWords)
Word2VecParam.Builder	useAdaGrad(boolean useAdaGrad)
Word2VecParam.Builder	weights(InMemoryLookupTable weights)
Word2VecParam.Builder	window(int window)
Word2VecParam.Builder	wordCount(org.apache.spark.broadcast.Broadcast<java.util.concurrent.atomic.AtomicLong> wordCount)

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Method Detail

expTable

public Word2VecParam.Builder expTable(org.apache.spark.broadcast.Broadcast<double[]> expTable)

useAdaGrad

public Word2VecParam.Builder useAdaGrad(boolean useAdaGrad)

negative

public Word2VecParam.Builder negative(double negative)

numWords

public Word2VecParam.Builder numWords(int numWords)

table

public Word2VecParam.Builder table(org.nd4j.linalg.api.ndarray.INDArray table)

window

public Word2VecParam.Builder window(int window)

setNextRandom

public Word2VecParam.Builder setNextRandom(java.util.concurrent.atomic.AtomicLong nextRandom)

setAlpha

public Word2VecParam.Builder setAlpha(double alpha)

setMinAlpha

public Word2VecParam.Builder setMinAlpha(double minAlpha)

totalWords

public Word2VecParam.Builder totalWords(int totalWords)

lastChecked

public Word2VecParam.Builder lastChecked(int lastChecked)

wordCount

public Word2VecParam.Builder wordCount(org.apache.spark.broadcast.Broadcast<java.util.concurrent.atomic.AtomicLong> wordCount)

weights

public Word2VecParam.Builder weights(InMemoryLookupTable weights)

setVectorLength

public Word2VecParam.Builder setVectorLength(int vectorLength)

build

public Word2VecParam build()