Pg2vecModelBuilder |
Pg2vecModelBuilder.setBatchSize(int batchSize) |
Sets the batch size for training the model (default: 128)
|
Pg2vecModelBuilder |
Pg2vecModelBuilder.setEnableAccelerator(boolean enableAccelerator) |
If true, use the accelerator (GPU) if available as the machine learning backend.
|
Pg2vecModelBuilder |
Pg2vecModelBuilder.setGraphLetIdPropertyName(java.lang.String graphLetIdName) |
Sets the property name of the graphlet-id in the input graph (required input)
|
Pg2vecModelBuilder |
Pg2vecModelBuilder.setGraphletSizePropertyName(java.lang.String graphletSizePropertyName) |
Sets the property name for graphlet size (default: "graphletSize-Pg2vec")
|
Pg2vecModelBuilder |
Pg2vecModelBuilder.setLayerSize(int layerSize) |
Sets the number of dimensions for the output vectors (default: 200)
|
Pg2vecModelBuilder |
Pg2vecModelBuilder.setLearningRate(double lr) |
Sets the initial learning rate (default: 0.025)
|
Pg2vecModelBuilder |
Pg2vecModelBuilder.setMinLearningRate(double minLr) |
Sets the minimum learning rate (default: 0.0001)
|
Pg2vecModelBuilder |
Pg2vecModelBuilder.setMinWordFrequency(int minWordFrequency) |
Sets the minimum word frequency to consider before pruning (default: 1)
|
Pg2vecModelBuilder |
Pg2vecModelBuilder.setNumEpochs(int numEpochs) |
Sets the number of epochs to train the model (default: 2)
|
Pg2vecModelBuilder |
Pg2vecModelBuilder.setSeed(java.lang.Long seed) |
Sets the random seed for training the model.
|
Pg2vecModelBuilder |
Pg2vecModelBuilder.setShuffle(boolean shuffle) |
Sets whether to shuffle the random walks before training on them
|
Pg2vecModelBuilder |
Pg2vecModelBuilder.setUseGraphletSize(boolean useGraphletSize) |
Sets whether to use or not the graphlet size (default: true)
|
Pg2vecModelBuilder |
Pg2vecModelBuilder.setVertexPropertyNames(java.util.Collection<java.lang.String> vertexPropertyNames) |
Sets the property names to consider for pg2vec model training (required input)
|
Pg2vecModelBuilder |
Pg2vecModelBuilder.setWalkLength(int walkLength) |
Sets the length of the walks (default: 5)
|
Pg2vecModelBuilder |
Pg2vecModelBuilder.setWalksPerVertex(int walksPerVertex) |
Sets the number of walks to consider per vertex (default: 4)
|
Pg2vecModelBuilder |
Pg2vecModelBuilder.setWindowSize(int windowSize) |
Sets the window size to consider while training the model (default: 5)
|