public class Pg2vecModelConfig extends java.lang.Object implements BaseModelConfig
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()
Gets the batch size
|
PropertyType |
getGraphIdPropertyType() |
java.lang.String |
getGraphLetIdPropertyName() |
java.lang.String |
getGraphletSizePropertyName() |
int |
getLayerSize() |
double |
getLearningRate() |
java.lang.Double |
getLoss() |
double |
getMinLearningRate() |
int |
getMinWordFrequency() |
int |
getNumEpochs()
Gets the number of epochs to train the model
|
java.lang.Long |
getSeed()
Gets the random seed
|
java.lang.Boolean |
getUseGraphletSize() |
double |
getValidationFraction()
Deprecated.
|
java.util.Collection<java.lang.String> |
getVertexPropertyNames() |
int |
getWalkLength() |
int |
getWalksPerVertex() |
int |
getWindowSize() |
boolean |
isShuffle()
Gets whether the model will shuffle or not
|
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)
Sets the loss after training
|
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)
Deprecated.
|
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()
BaseModelConfig
getBatchSize
in interface BaseModelConfig
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()
BaseModelConfig
getNumEpochs
in interface BaseModelConfig
public java.lang.Long getSeed()
BaseModelConfig
getSeed
in interface BaseModelConfig
public java.lang.Boolean getUseGraphletSize()
@Deprecated public double getValidationFraction()
public java.util.Collection<java.lang.String> getVertexPropertyNames()
public int getWalkLength()
public int getWalksPerVertex()
public int getWindowSize()
public boolean isShuffle()
BaseModelConfig
isShuffle
in interface BaseModelConfig
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)
BaseModelConfig
setLoss
in interface BaseModelConfig
trainingLoss
- the value of the training losspublic 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)
@Deprecated 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)