Class DeepWalk.Builder<V,E>
- java.lang.Object
-
- org.deeplearning4j.graph.models.deepwalk.DeepWalk.Builder<V,E>
-
-
Constructor Summary
Constructors Constructor Description Builder()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description DeepWalk<V,E>
build()
DeepWalk.Builder<V,E>
learningRate(double learningRate)
Set the learning rateDeepWalk.Builder<V,E>
seed(long seed)
Seed for random number generation (used for repeatability).DeepWalk.Builder<V,E>
vectorSize(int vectorSize)
Sets the size of the vectors to be learned for each vertex in the graphDeepWalk.Builder<V,E>
windowSize(int windowSize)
Sets the window size used in skipgram model
-
-
-
Method Detail
-
vectorSize
public DeepWalk.Builder<V,E> vectorSize(int vectorSize)
Sets the size of the vectors to be learned for each vertex in the graph
-
learningRate
public DeepWalk.Builder<V,E> learningRate(double learningRate)
Set the learning rate
-
windowSize
public DeepWalk.Builder<V,E> windowSize(int windowSize)
Sets the window size used in skipgram model
-
seed
public DeepWalk.Builder<V,E> seed(long seed)
Seed for random number generation (used for repeatability). Note however that parallel/async gradient descent might result in behaviour that is not repeatable, in spite of setting seed
-
-