Package oracle.pgx.config.mllib
Class SupervisedGraphWiseModelConfig
- java.lang.Object
-
- oracle.pgx.config.mllib.GraphWiseBaseModelConfig
-
- oracle.pgx.config.mllib.GraphWiseModelConfig
-
- oracle.pgx.config.mllib.SupervisedGraphWiseModelConfig
-
public class SupervisedGraphWiseModelConfig extends GraphWiseModelConfig
Configuration class forSupervisedGraphWiseModel
. SeeSupervisedGraphWiseModel
for a description of the hyperparameters.- Since:
- 19.4
-
-
Nested Class Summary
-
Nested classes/interfaces inherited from class oracle.pgx.config.mllib.GraphWiseModelConfig
GraphWiseModelConfig.GraphConvModelVariant
-
Nested classes/interfaces inherited from class oracle.pgx.config.mllib.GraphWiseBaseModelConfig
GraphWiseBaseModelConfig.Backend
-
-
Field Summary
Fields Modifier and Type Field Description static BatchGenerator
DEFAULT_BATCH_GENERATOR
static java.util.Map<?,java.lang.Float>
DEFAULT_CLASS_MAP
nullstatic LossFunction
DEFAULT_LOSS_FUNCTION_CLASS
static GraphWisePredictionLayerConfig[]
DEFAULT_PREDICTION_LAYER_CONFIGS
one default initialized config (SeeGraphWisePredictionLayerConfig
)static java.util.EnumSet<PropertyType>
SUPPORTED_LABEL_TYPES
INTEGER, STRING, BOOLEAN, LONGstatic java.util.EnumSet<PropertyType>
SUPPORTED_REGRESSION_TYPES
FLOAT, DOUBLE, INTEGER, LONG-
Fields inherited from class oracle.pgx.config.mllib.GraphWiseModelConfig
DEFAULT_MODE
-
Fields inherited from class oracle.pgx.config.mllib.GraphWiseBaseModelConfig
DEFAULT_BACKEND, DEFAULT_BATCH_SIZE, DEFAULT_CONV_LAYER_CONFIGS, DEFAULT_EMBEDDING_DIM, DEFAULT_ENABLE_ACCELERATOR, DEFAULT_LEARNING_RATE, DEFAULT_NORMALIZE, DEFAULT_NUM_EPOCHS, DEFAULT_SEED, DEFAULT_SHUFFLE, DEFAULT_STANDARDIZE, DEFAULT_WEIGHT_DECAY, SUPPORTED_INPUT_TYPES
-
-
Constructor Summary
Constructors Constructor Description SupervisedGraphWiseModelConfig()
SupervisedGraphWiseModelConfig(int batchSize, int numEpochs, double learningRate, double weightDecay, int embeddingDim, java.lang.Integer seed, GraphWiseBaseConvLayerConfig[] convLayerConfigs, boolean standardize, boolean normalize, boolean shuffle, java.util.List<java.lang.String> vertexInputPropertyNames, java.util.List<java.lang.String> edgeInputPropertyNames, java.util.Map<java.lang.String,InputPropertyConfig> vertexInputPropertyConfigs, java.util.Map<java.lang.String,InputPropertyConfig> edgeInputPropertyConfigs, oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig, java.util.List<java.util.Set<java.lang.String>> targetVertexLabelSets, boolean fitted, double trainingLoss, int inputFeatureDim, int edgeInputFeatureDim, oracle.pgx.config.mllib.SupervisedGraphWiseModelConfig.LossFunction lossFunction, LossFunction lossFunctionClass, BatchGenerator batchGenerator, GraphWisePredictionLayerConfig[] predictionLayerConfigs, java.lang.String vertexTargetPropertyName, LabelMaps labelMaps, GraphWiseBaseModelConfig.Backend backend, GraphWiseModelConfig.GraphConvModelVariant variant, boolean enableAccelerator, GraphWiseValidationConfig validationConfig)
SupervisedGraphWiseModelConfig(SupervisedGraphWiseModelConfig source)
SupervisedGraphWiseModelConfig(SupervisedGraphWiseModelConfig source, oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description BatchGenerator
getBatchGenerator()
java.util.Map<?,java.lang.Integer>
getClassMap()
java.util.Map<?,java.lang.Float>
getClassWeights()
LabelMaps
getLabelMaps()
PropertyType
getLabelType()
LossFunction
getLossFunctionClass()
int
getNumClasses()
GraphWisePredictionLayerConfig[]
getPredictionLayerConfigs()
java.lang.String
getVertexTargetPropertyName()
void
setBatchGenerator(BatchGenerator batchGenerator)
void
setClassMap(java.util.Map<?,java.lang.Integer> classMap)
void
setClassWeights(java.util.Map<?,java.lang.Float> classWeights)
void
setLabelMaps(LabelMaps labelMaps)
void
setLabelType(PropertyType labelType)
void
setLossFunctionClass(LossFunction lossFunction)
void
setPredictionLayerConfigs(GraphWisePredictionLayerConfig... predictionLayerConfigs)
void
setVertexTargetPropertyName(java.lang.String vertexTargetPropertyName)
-
Methods inherited from class oracle.pgx.config.mllib.GraphWiseModelConfig
getTargetVertexLabelSets, getVariant, setTargetVertexLabels, setTargetVertexLabelSets, setVariant
-
Methods inherited from class oracle.pgx.config.mllib.GraphWiseBaseModelConfig
clearSensitiveData, getBackend, getBatchSize, getCategoryMappingConfig, getConvLayerConfigs, getEdgeInputFeatureDim, getEdgeInputPropertyConfigs, getEdgeInputPropertyNames, getEmbeddingDim, getInputFeatureDim, getLearningRate, getNumEpochs, getSeed, getTrainingLoss, getValidationConfig, getVertexInputPropertyConfigs, getVertexInputPropertyNames, getWeightDecay, isEnableAccelerator, isFitted, isNormalize, isShuffle, isStandardize, setBatchSize, setCategoryMappingConfig, setConvLayerConfigs, setEdgeInputFeatureDim, setEdgeInputPropertyConfigs, setEdgeInputPropertyNames, setEmbeddingDim, setEnableAccelerator, setFitted, setInputFeatureDim, setLearningRate, setNormalize, setNumEpochs, setSeed, setShuffle, setStandardize, setTrainingLoss, setValidationConfig, setVertexInputPropertyConfigs, setVertexInputPropertyNames, setWeightDecay
-
-
-
-
Field Detail
-
SUPPORTED_LABEL_TYPES
public static final java.util.EnumSet<PropertyType> SUPPORTED_LABEL_TYPES
INTEGER, STRING, BOOLEAN, LONG
-
SUPPORTED_REGRESSION_TYPES
public static final java.util.EnumSet<PropertyType> SUPPORTED_REGRESSION_TYPES
FLOAT, DOUBLE, INTEGER, LONG
-
DEFAULT_LOSS_FUNCTION_CLASS
public static final LossFunction DEFAULT_LOSS_FUNCTION_CLASS
-
DEFAULT_BATCH_GENERATOR
public static final BatchGenerator DEFAULT_BATCH_GENERATOR
-
DEFAULT_PREDICTION_LAYER_CONFIGS
public static final GraphWisePredictionLayerConfig[] DEFAULT_PREDICTION_LAYER_CONFIGS
one default initialized config (SeeGraphWisePredictionLayerConfig
)
-
DEFAULT_CLASS_MAP
public static final java.util.Map<?,java.lang.Float> DEFAULT_CLASS_MAP
null
-
-
Constructor Detail
-
SupervisedGraphWiseModelConfig
public SupervisedGraphWiseModelConfig()
-
SupervisedGraphWiseModelConfig
public SupervisedGraphWiseModelConfig(int batchSize, int numEpochs, double learningRate, double weightDecay, int embeddingDim, java.lang.Integer seed, GraphWiseBaseConvLayerConfig[] convLayerConfigs, boolean standardize, boolean normalize, boolean shuffle, java.util.List<java.lang.String> vertexInputPropertyNames, java.util.List<java.lang.String> edgeInputPropertyNames, java.util.Map<java.lang.String,InputPropertyConfig> vertexInputPropertyConfigs, java.util.Map<java.lang.String,InputPropertyConfig> edgeInputPropertyConfigs, oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig, java.util.List<java.util.Set<java.lang.String>> targetVertexLabelSets, boolean fitted, double trainingLoss, int inputFeatureDim, int edgeInputFeatureDim, oracle.pgx.config.mllib.SupervisedGraphWiseModelConfig.LossFunction lossFunction, LossFunction lossFunctionClass, BatchGenerator batchGenerator, GraphWisePredictionLayerConfig[] predictionLayerConfigs, java.lang.String vertexTargetPropertyName, LabelMaps labelMaps, GraphWiseBaseModelConfig.Backend backend, GraphWiseModelConfig.GraphConvModelVariant variant, boolean enableAccelerator, GraphWiseValidationConfig validationConfig)
-
SupervisedGraphWiseModelConfig
public SupervisedGraphWiseModelConfig(SupervisedGraphWiseModelConfig source)
-
SupervisedGraphWiseModelConfig
public SupervisedGraphWiseModelConfig(SupervisedGraphWiseModelConfig source, oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig)
-
-
Method Detail
-
getNumClasses
public int getNumClasses()
-
getVertexTargetPropertyName
public java.lang.String getVertexTargetPropertyName()
-
setVertexTargetPropertyName
public final void setVertexTargetPropertyName(java.lang.String vertexTargetPropertyName)
-
getPredictionLayerConfigs
public GraphWisePredictionLayerConfig[] getPredictionLayerConfigs()
-
setPredictionLayerConfigs
public final void setPredictionLayerConfigs(GraphWisePredictionLayerConfig... predictionLayerConfigs)
-
getLossFunctionClass
public LossFunction getLossFunctionClass()
-
setLossFunctionClass
public final void setLossFunctionClass(LossFunction lossFunction)
-
getBatchGenerator
public BatchGenerator getBatchGenerator()
-
setBatchGenerator
public final void setBatchGenerator(BatchGenerator batchGenerator)
-
getClassMap
public java.util.Map<?,java.lang.Integer> getClassMap()
-
setClassMap
public final void setClassMap(java.util.Map<?,java.lang.Integer> classMap)
-
setClassWeights
public final void setClassWeights(java.util.Map<?,java.lang.Float> classWeights)
-
getClassWeights
public java.util.Map<?,java.lang.Float> getClassWeights()
-
getLabelType
public PropertyType getLabelType()
-
setLabelType
public final void setLabelType(PropertyType labelType)
-
getLabelMaps
public LabelMaps getLabelMaps()
-
setLabelMaps
public final void setLabelMaps(LabelMaps labelMaps)
-
-