public class Pg2vecModelConfig
extends java.lang.Object
Constructor and Description |
---|
Pg2vecModelConfig() |
Pg2vecModelConfig(int walkLength, int walksPerVertex, int minWordFrequency, int batchSize, int numEpochs, int layerSize, double learningRate, double minLearningRate, int windowSize, java.util.Collection<java.lang.String> vertexPropertyNames, java.lang.String graphLetIdName, PropertyType graphIdPropertyType, java.lang.Double trainingLoss, double validationFraction, boolean useGraphletSize, java.lang.Long seed, boolean shuffle, java.lang.String graphletSizePropertyName) |
Modifier and Type | Method and Description |
---|---|
int |
getBatchSize() |
PropertyType |
getGraphIdPropertyType() |
java.lang.String |
getGraphLetIdPropertyName() |
java.lang.String |
getGraphletSizePropertyName() |
int |
getLayerSize() |
double |
getLearningRate() |
java.lang.Double |
getLoss() |
double |
getMinLearningRate() |
int |
getMinWordFrequency() |
int |
getNumEpochs() |
java.lang.Long |
getSeed() |
java.lang.Boolean |
getUseGraphletSize() |
double |
getValidationFraction() |
java.util.Collection<java.lang.String> |
getVertexPropertyNames() |
int |
getWalkLength() |
int |
getWalksPerVertex() |
int |
getWindowSize() |
boolean |
isShuffle() |
void |
setBatchSize(int batchSize) |
void |
setGraphIdPropertyType(PropertyType graphIdPropertyType) |
void |
setGraphLetIdPropertyName(java.lang.String graphLetIdName) |
void |
setGraphletSizePropertyName(java.lang.String graphletSizePropertyName) |
void |
setLayerSize(int layerSize) |
void |
setLearningRate(double learningRate) |
void |
setLoss(java.lang.Double trainingLoss) |
void |
setMinLearningRate(double minLearningRate) |
void |
setMinWordFrequency(int minWordFrequency) |
void |
setNumEpochs(int numEpochs) |
void |
setSeed(java.lang.Long seed) |
void |
setShuffle(boolean shuffle) |
void |
setUseGraphletSize(java.lang.Boolean useGraphletSize) |
void |
setValidationFraction(double validationFraction) |
void |
setVertexPropertyNames(java.util.Collection<java.lang.String> vertexPropertyNames) |
void |
setWalkLength(int walkLength) |
void |
setWalksPerVertex(int walksPerVertex) |
void |
setWindowSize(int windowSize) |
public Pg2vecModelConfig()
public Pg2vecModelConfig(int walkLength, int walksPerVertex, int minWordFrequency, int batchSize, int numEpochs, int layerSize, double learningRate, double minLearningRate, int windowSize, java.util.Collection<java.lang.String> vertexPropertyNames, java.lang.String graphLetIdName, PropertyType graphIdPropertyType, java.lang.Double trainingLoss, double validationFraction, boolean useGraphletSize, java.lang.Long seed, boolean shuffle, java.lang.String graphletSizePropertyName)
public int getBatchSize()
public PropertyType getGraphIdPropertyType()
public java.lang.String getGraphLetIdPropertyName()
public java.lang.String getGraphletSizePropertyName()
public int getLayerSize()
public double getLearningRate()
public java.lang.Double getLoss()
public double getMinLearningRate()
public int getMinWordFrequency()
public int getNumEpochs()
public java.lang.Long getSeed()
public java.lang.Boolean getUseGraphletSize()
public double getValidationFraction()
public java.util.Collection<java.lang.String> getVertexPropertyNames()
public int getWalkLength()
public int getWalksPerVertex()
public int getWindowSize()
public boolean isShuffle()
public void setBatchSize(int batchSize)
public void setGraphIdPropertyType(PropertyType graphIdPropertyType)
public void setGraphLetIdPropertyName(java.lang.String graphLetIdName)
public void setGraphletSizePropertyName(java.lang.String graphletSizePropertyName)
public void setLayerSize(int layerSize)
public void setLearningRate(double learningRate)
public void setLoss(java.lang.Double trainingLoss)
public void setMinLearningRate(double minLearningRate)
public void setMinWordFrequency(int minWordFrequency)
public void setNumEpochs(int numEpochs)
public void setSeed(java.lang.Long seed)
public void setShuffle(boolean shuffle)
public void setUseGraphletSize(java.lang.Boolean useGraphletSize)
public void setValidationFraction(double validationFraction)
public void setVertexPropertyNames(java.util.Collection<java.lang.String> vertexPropertyNames)
public void setWalkLength(int walkLength)
public void setWalksPerVertex(int walksPerVertex)
public void setWindowSize(int windowSize)