DeepWalkModelBuilder |
DeepWalkModelBuilder.setBatchSize(int batchSize) |
Sets the batch size for training the model (default: 128)
|
DeepWalkModelBuilder |
DeepWalkModelBuilder.setEnableAccelerator(boolean enableAccelerator) |
If true, use the accelerator (GPU) if available as the machine learning backend.
|
DeepWalkModelBuilder |
DeepWalkModelBuilder.setIgnoreIsolated(boolean ignoreIsolated) |
Whether to ignore isolated vertices.
|
DeepWalkModelBuilder |
DeepWalkModelBuilder.setLayerSize(int layerSize) |
Sets the number of dimensions for the output vectors (default: 200)
|
DeepWalkModelBuilder |
DeepWalkModelBuilder.setLearningRate(double lr) |
Sets the initial learning rate (default: 0.025)
|
DeepWalkModelBuilder |
DeepWalkModelBuilder.setMinLearningRate(double minLr) |
Sets the minimum learning rate (default: 0.0001)
|
DeepWalkModelBuilder |
DeepWalkModelBuilder.setMinWordFrequency(int minWordFrequency) |
Sets the minimum word frequency to consider before pruning (default: 1)
|
DeepWalkModelBuilder |
DeepWalkModelBuilder.setNegativeSample(int negativeSample) |
Sets the number of negative samples (default: 10)
|
DeepWalkModelBuilder |
DeepWalkModelBuilder.setNumEpochs(int numEpochs) |
Sets the number of epochs to train the model (default: 2)
|
DeepWalkModelBuilder |
DeepWalkModelBuilder.setSampleRate(double sampleRate) |
Sets the sample rate (default: 0.0)
Used to subsample frequent nodes.
|
DeepWalkModelBuilder |
DeepWalkModelBuilder.setSeed(java.lang.Long seed) |
Sets the random seed for training the model.
|
DeepWalkModelBuilder |
DeepWalkModelBuilder.setShuffle(boolean shuffle) |
Sets whether to shuffle the random walks before training on them
|
DeepWalkModelBuilder |
DeepWalkModelBuilder.setWalkLength(int walkLength) |
Sets the length of the walks (default: 5)
|
DeepWalkModelBuilder |
DeepWalkModelBuilder.setWalksPerVertex(int walksPerVertex) |
Sets the number of walks to consider per vertex (default: 4)
|
DeepWalkModelBuilder |
DeepWalkModelBuilder.setWindowSize(int windowSize) |
Sets the window size to consider while training the model (default: 5)
|