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 |
---|---|
GraphWiseConvLayerConfig |
build()
Build the configuration
|
GraphWiseConvLayerConfigBuilder |
setActivationFunction(ActivationFunction activationFunction)
Set the activation function for this layer (see
ActivationFunction ) |
GraphWiseConvLayerConfigBuilder |
setDropoutRate(double dropoutRate)
Set the dropout rate for this layer.
|
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(WeightInitScheme weightInitScheme)
Set the weight initialization scheme for the weights of this layer.
|
GraphWiseConvLayerConfigBuilder |
useEdgeToEdgeConnection(boolean edgeToEdgeConnection)
Use the connection between edges and edges.
|
GraphWiseConvLayerConfigBuilder |
useEdgeToVertexConnection(boolean edgeToVertexConnection)
Use the connection between edges and vertices.
|
GraphWiseConvLayerConfigBuilder |
useVertexToEdgeConnection(boolean vertexToEdgeConnection)
Use the connection between vertices and edges.
|
GraphWiseConvLayerConfigBuilder |
useVertexToVertexConnection(boolean vertexToVertexConnection)
Use the connection between vertices and vertices.
|
public GraphWiseConvLayerConfig build()
public GraphWiseConvLayerConfigBuilder setActivationFunction(ActivationFunction activationFunction)
ActivationFunction
)
default: GraphWiseConvLayerConfig.DEFAULT_ACTIVATION_FUNCTION
activationFunction
- activation functionpublic GraphWiseConvLayerConfigBuilder setDropoutRate(double dropoutRate)
dropoutRate
- dropout ratepublic 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(WeightInitScheme weightInitScheme)
WeightInitScheme
.
default: GraphWiseConvLayerConfig.DEFAULT_WEIGHT_INIT_SCHEME
weightInitScheme
- weight initialization schemepublic GraphWiseConvLayerConfigBuilder useEdgeToEdgeConnection(boolean edgeToEdgeConnection)
default: GraphWiseConvLayerConfig#DEFAULT_EDGE_EDGE_CONNECTION
edgeToEdgeConnection
- boolean flag to use edge to edge connectionpublic GraphWiseConvLayerConfigBuilder useEdgeToVertexConnection(boolean edgeToVertexConnection)
default: GraphWiseConvLayerConfig#DEFAULT_EDGE_VERTEX_CONNECTION
edgeToVertexConnection
- boolean flag to use edge to vertex connectionpublic GraphWiseConvLayerConfigBuilder useVertexToEdgeConnection(boolean vertexToEdgeConnection)
default: GraphWiseConvLayerConfig#DEFAULT_VERTEX_EDGE_CONNECTION
vertexToEdgeConnection
- boolean flag to use vertex to edge connectionpublic GraphWiseConvLayerConfigBuilder useVertexToVertexConnection(boolean vertexToVertexConnection)
default: GraphWiseConvLayerConfig#DEFAULT_VERTEX_VERTEX_CONNECTION
vertexToVertexConnection
- boolean flag to use vertex to vertex connection