Package | Description |
---|---|
org.deeplearning4j.graph.models.deepwalk |
Modifier and Type | Method and Description |
---|---|
DeepWalk.Builder<V,E> |
DeepWalk.Builder.learningRate(double learningRate)
Set the learning rate
|
DeepWalk.Builder<V,E> |
DeepWalk.Builder.seed(long seed)
Seed for random number generation (used for repeatability).
|
DeepWalk.Builder<V,E> |
DeepWalk.Builder.vectorSize(int vectorSize)
Sets the size of the vectors to be learned for each vertex in the graph
|
DeepWalk.Builder<V,E> |
DeepWalk.Builder.windowSize(int windowSize)
Sets the window size used in skipgram model
|
Copyright © 2020. All rights reserved.