public class GraphWiseConvLayerConfigBuilder
extends java.lang.Object
GraphWiseConvLayerConfig
. Currently, only mean aggregation layers are supported.Constructor and Description |
---|
GraphWiseConvLayerConfigBuilder() |
Modifier and Type | Method and Description |
---|---|
oracle.pgx.config.mllib.GraphWiseConvLayerConfig |
build()
Build the configuration
|
GraphWiseConvLayerConfigBuilder |
setActivationFunction(oracle.pgx.config.mllib.ActivationFunction activationFunction)
Set the activation function for this layer (see
ActivationFunction ) |
GraphWiseConvLayerConfigBuilder |
setNumSampledNeighbors(int numSampledNeighbors)
Set the number of neighbors that will be sampled in this layer.
|
GraphWiseConvLayerConfigBuilder |
setWeightedAggregationProperty(java.lang.String neighborWeightProperty)
Set the graph property to be used for weighted aggregation.
|
GraphWiseConvLayerConfigBuilder |
setWeightInitScheme(oracle.pgx.config.mllib.WeightInitScheme weightInitScheme)
Set the weight initialization scheme for the weights of this layer.
|
public oracle.pgx.config.mllib.GraphWiseConvLayerConfig build()
public GraphWiseConvLayerConfigBuilder setActivationFunction(oracle.pgx.config.mllib.ActivationFunction activationFunction)
ActivationFunction
)
default: GraphWiseConvLayerConfig.DEFAULT_ACTIVATION_FUNCTION
activationFunction
- activation functionpublic GraphWiseConvLayerConfigBuilder setNumSampledNeighbors(int numSampledNeighbors)
default: GraphWiseConvLayerConfig.DEFAULT_NUM_SAMPLED_NEIGHBORS
numSampledNeighbors
- number of neighbors to samplepublic GraphWiseConvLayerConfigBuilder setWeightedAggregationProperty(java.lang.String neighborWeightProperty)
Weighted aggregation will perform the mean aggregation of neighbors based on weights given by the specified graph property.
default: GraphWiseConvLayerConfig.DEFAULT_NEIGHBOR_WEIGHT_PROPERTY_NAME
neighborWeightProperty
- the property namepublic GraphWiseConvLayerConfigBuilder setWeightInitScheme(oracle.pgx.config.mllib.WeightInitScheme weightInitScheme)
WeightInitScheme
.
default: GraphWiseConvLayerConfig.DEFAULT_WEIGHT_INIT_SCHEME
weightInitScheme
- weight initialization scheme