public class SupervisedGraphWiseModelConfig extends GraphWiseModelConfig
SupervisedGraphWiseModel
.
See SupervisedGraphWiseModel
for a description of the hyperparameters.Modifier and Type | Class and Description |
---|---|
static class |
oracle.pgx.config.mllib.SupervisedGraphWiseModelConfig.LossFunction
Deprecated.
|
GraphWiseModelConfig.GraphConvModelVariant
GraphWiseBaseModelConfig.Backend
Modifier and Type | Field and Description |
---|---|
static BatchGenerator |
DEFAULT_BATCH_GENERATOR
|
static java.util.Map<?,java.lang.Float> |
DEFAULT_CLASS_MAP
null
|
static oracle.pgx.config.mllib.SupervisedGraphWiseModelConfig.LossFunction |
DEFAULT_LOSS_FUNCTION
Deprecated.
|
static LossFunction |
DEFAULT_LOSS_FUNCTION_CLASS
|
static GraphWisePredictionLayerConfig[] |
DEFAULT_PREDICTION_LAYER_CONFIGS
one default initialized config (See
GraphWisePredictionLayerConfig ) |
static java.util.EnumSet<PropertyType> |
SUPPORTED_LABEL_TYPES
INTEGER, STRING, BOOLEAN, LONG
|
static java.util.EnumSet<PropertyType> |
SUPPORTED_REGRESSION_TYPES
FLOAT, DOUBLE, INTEGER, LONG
|
DEFAULT_MODE
DEFAULT_BACKEND, DEFAULT_BATCH_SIZE, DEFAULT_CONV_LAYER_CONFIGS, DEFAULT_EMBEDDING_DIM, DEFAULT_LEARNING_RATE, DEFAULT_NORMALIZE, DEFAULT_NUM_EPOCHS, DEFAULT_SEED, DEFAULT_SHUFFLE, DEFAULT_STANDARDIZE, DEFAULT_WEIGHT_DECAY, SUPPORTED_INPUT_TYPES
Constructor and 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) |
SupervisedGraphWiseModelConfig(SupervisedGraphWiseModelConfig source) |
SupervisedGraphWiseModelConfig(SupervisedGraphWiseModelConfig source,
oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig) |
Modifier and Type | Method and Description |
---|---|
BatchGenerator |
getBatchGenerator() |
java.util.Map<?,java.lang.Integer> |
getClassMap() |
java.util.Map<?,java.lang.Float> |
getClassWeights() |
LabelMaps |
getLabelMaps() |
PropertyType |
getLabelType() |
oracle.pgx.config.mllib.SupervisedGraphWiseModelConfig.LossFunction |
getLossFunction()
Deprecated.
|
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 |
setLossFunction(oracle.pgx.config.mllib.SupervisedGraphWiseModelConfig.LossFunction lossFunction)
Deprecated.
|
void |
setLossFunctionClass(LossFunction lossFunction) |
void |
setPredictionLayerConfigs(GraphWisePredictionLayerConfig... predictionLayerConfigs) |
void |
setVertexTargetPropertyName(java.lang.String vertexTargetPropertyName) |
getTargetVertexLabelSets, getVariant, setTargetVertexLabels, setTargetVertexLabelSets, setVariant
clearSensitiveData, getBackend, getBatchSize, getCategoryMappingConfig, getConvLayerConfigs, getEdgeInputFeatureDim, getEdgeInputPropertyConfigs, getEdgeInputPropertyNames, getEmbeddingDim, getInputFeatureDim, getLearningRate, getNumEpochs, getSeed, getTrainingLoss, getVertexInputPropertyConfigs, getVertexInputPropertyNames, getWeightDecay, isFitted, isNormalize, isShuffle, isStandardize, setBatchSize, setCategoryMappingConfig, setConvLayerConfigs, setEdgeInputFeatureDim, setEdgeInputPropertyConfigs, setEdgeInputPropertyNames, setEmbeddingDim, setFitted, setInputFeatureDim, setLearningRate, setNormalize, setNumEpochs, setSeed, setShuffle, setStandardize, setTrainingLoss, setVertexInputPropertyConfigs, setVertexInputPropertyNames, setWeightDecay
public static final BatchGenerator DEFAULT_BATCH_GENERATOR
public static final java.util.Map<?,java.lang.Float> DEFAULT_CLASS_MAP
@Deprecated public static final oracle.pgx.config.mllib.SupervisedGraphWiseModelConfig.LossFunction DEFAULT_LOSS_FUNCTION
SupervisedGraphWiseModelConfig.LossFunction.SOFTMAX_CROSS_ENTROPY
public static final LossFunction DEFAULT_LOSS_FUNCTION_CLASS
public static final GraphWisePredictionLayerConfig[] DEFAULT_PREDICTION_LAYER_CONFIGS
GraphWisePredictionLayerConfig
)public static final java.util.EnumSet<PropertyType> SUPPORTED_LABEL_TYPES
public static final java.util.EnumSet<PropertyType> SUPPORTED_REGRESSION_TYPES
public 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)
public SupervisedGraphWiseModelConfig(SupervisedGraphWiseModelConfig source)
public SupervisedGraphWiseModelConfig(SupervisedGraphWiseModelConfig source, oracle.pgx.config.internal.categorymapping.CategoryMappingConfig categoryMappingConfig)
public BatchGenerator getBatchGenerator()
public java.util.Map<?,java.lang.Integer> getClassMap()
public java.util.Map<?,java.lang.Float> getClassWeights()
public LabelMaps getLabelMaps()
public PropertyType getLabelType()
@Deprecated public oracle.pgx.config.mllib.SupervisedGraphWiseModelConfig.LossFunction getLossFunction()
public LossFunction getLossFunctionClass()
public int getNumClasses()
public GraphWisePredictionLayerConfig[] getPredictionLayerConfigs()
public java.lang.String getVertexTargetPropertyName()
public final void setBatchGenerator(BatchGenerator batchGenerator)
public final void setClassMap(java.util.Map<?,java.lang.Integer> classMap)
public final void setClassWeights(java.util.Map<?,java.lang.Float> classWeights)
public final void setLabelMaps(LabelMaps labelMaps)
public final void setLabelType(PropertyType labelType)
@Deprecated public final void setLossFunction(oracle.pgx.config.mllib.SupervisedGraphWiseModelConfig.LossFunction lossFunction)
public final void setLossFunctionClass(LossFunction lossFunction)
public final void setPredictionLayerConfigs(GraphWisePredictionLayerConfig... predictionLayerConfigs)
public final void setVertexTargetPropertyName(java.lang.String vertexTargetPropertyName)