| Package | Description |
|---|---|
| oracle.pgx.config.mllib |
This package contains APIs to all graph machine learning features of PGX.
|
| Modifier and Type | Field and Description |
|---|---|
static GraphWiseModelConfig.Backend |
GraphWiseModelConfig.DEFAULT_BACKEND
LibTorch
|
| Modifier and Type | Method and Description |
|---|---|
GraphWiseModelConfig.Backend |
GraphWiseModelConfig.getBackend() |
static GraphWiseModelConfig.Backend |
GraphWiseModelConfig.Backend.valueOf(java.lang.String name)
Returns the enum constant of this type with the specified name.
|
static GraphWiseModelConfig.Backend[] |
GraphWiseModelConfig.Backend.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
| Constructor and Description |
|---|
GraphWiseModelConfig(int batchSize,
int numEpochs,
double learningRate,
double weightDecay,
int embeddingDim,
java.lang.Integer seed,
GraphWiseConvLayerConfig[] convLayerConfigs,
boolean standardize,
boolean shuffle,
java.util.List<java.lang.String> vertexInputPropertyNames,
java.util.List<java.lang.String> edgeInputPropertyNames,
boolean fitted,
double trainingLoss,
int vertexInputFeatureDim,
int edgeInputFeatureDim,
java.util.List<java.util.Set<java.lang.String>> targetVertexLabelSets,
GraphWiseModelConfig.Backend backend,
GraphWiseModelConfig.GraphConvModelVariant variant) |
SupervisedGraphWiseModelConfig(int batchSize,
int numEpochs,
double learningRate,
double weightDecay,
int embeddingDim,
java.lang.Integer seed,
GraphWiseConvLayerConfig[] convLayerConfigs,
boolean standardize,
boolean shuffle,
java.util.List<java.lang.String> vertexInputPropertyNames,
java.util.List<java.lang.String> edgeInputPropertyNames,
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,
boolean normalize,
java.lang.String vertexTargetPropertyName,
LabelMaps labelMaps,
GraphWiseModelConfig.Backend backend,
GraphWiseModelConfig.GraphConvModelVariant variant) |
UnsupervisedGraphWiseModelConfig(int batchSize,
int numEpochs,
double learningRate,
double weightDecay,
int embeddingDim,
java.lang.Integer seed,
GraphWiseConvLayerConfig[] convLayerConfigs,
boolean standardize,
boolean shuffle,
java.util.List<java.lang.String> vertexInputPropertyNames,
java.util.List<java.lang.String> edgeInputPropertyNames,
java.util.List<java.util.Set<java.lang.String>> targetVertexLabels,
boolean fitted,
double trainingLoss,
int inputFeatureDim,
int inputEdgeFeatureDim,
UnsupervisedGraphWiseModelConfig.LossFunction lossFunction,
GraphWiseDgiLayerConfig dgiLayerConfig,
GraphWiseModelConfig.Backend backend,
GraphWiseModelConfig.GraphConvModelVariant variant) |