public static class Word2VecParam.Builder extends Object
| Constructor and Description |
|---|
Builder() |
| 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(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<AtomicLong> wordCount) |
public Word2VecParam.Builder expTable(org.apache.spark.broadcast.Broadcast<double[]> expTable)
public Word2VecParam.Builder useAdaGrad(boolean useAdaGrad)
public Word2VecParam.Builder negative(double negative)
public Word2VecParam.Builder numWords(int numWords)
public Word2VecParam.Builder table(org.nd4j.linalg.api.ndarray.INDArray table)
public Word2VecParam.Builder window(int window)
public Word2VecParam.Builder setNextRandom(AtomicLong nextRandom)
public Word2VecParam.Builder setAlpha(double alpha)
public Word2VecParam.Builder setMinAlpha(double minAlpha)
public Word2VecParam.Builder totalWords(int totalWords)
public Word2VecParam.Builder lastChecked(int lastChecked)
public Word2VecParam.Builder wordCount(org.apache.spark.broadcast.Broadcast<AtomicLong> wordCount)
public Word2VecParam.Builder weights(InMemoryLookupTable weights)
public Word2VecParam.Builder setVectorLength(int vectorLength)
public Word2VecParam build()
Copyright © 2020. All rights reserved.