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