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. SeeSupervisedGraphWiseModelfor 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 BatchGeneratorDEFAULT_BATCH_GENERATORstatic java.util.Map<?,java.lang.Float>DEFAULT_CLASS_MAPnullstatic LossFunctionDEFAULT_LOSS_FUNCTION_CLASSstatic GraphWisePredictionLayerConfig[]DEFAULT_PREDICTION_LAYER_CONFIGSone default initialized config (SeeGraphWisePredictionLayerConfig)static java.util.EnumSet<PropertyType>SUPPORTED_LABEL_TYPESINTEGER, STRING, BOOLEAN, LONGstatic java.util.EnumSet<PropertyType>SUPPORTED_REGRESSION_TYPESFLOAT, 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, 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 BatchGeneratorgetBatchGenerator()java.util.Map<?,java.lang.Integer>getClassMap()java.util.Map<?,java.lang.Float>getClassWeights()LabelMapsgetLabelMaps()PropertyTypegetLabelType()LossFunctiongetLossFunctionClass()intgetNumClasses()GraphWisePredictionLayerConfig[]getPredictionLayerConfigs()java.lang.StringgetVertexTargetPropertyName()voidsetBatchGenerator(BatchGenerator batchGenerator)voidsetClassMap(java.util.Map<?,java.lang.Integer> classMap)voidsetClassWeights(java.util.Map<?,java.lang.Float> classWeights)voidsetLabelMaps(LabelMaps labelMaps)voidsetLabelType(PropertyType labelType)voidsetLossFunctionClass(LossFunction lossFunction)voidsetPredictionLayerConfigs(GraphWisePredictionLayerConfig... predictionLayerConfigs)voidsetVertexTargetPropertyName(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, 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)
-
-