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) |