public abstract class GraphWiseBaseConvLayerConfigBuilder<Config extends GraphWiseBaseConvLayerConfig,Self extends GraphWiseBaseConvLayerConfigBuilder>
extends java.lang.Object
GraphWiseBaseConvLayerConfig
.Constructor and Description |
---|
GraphWiseBaseConvLayerConfigBuilder() |
Modifier and Type | Method and Description |
---|---|
Self |
setActivationFunction(ActivationFunction activationFunction)
Set the activation function for this layer (see
ActivationFunction ) |
Self |
setDropoutRate(double dropoutRate)
Set the dropout rate for this layer.
|
Self |
setNumSampledNeighbors(int numSampledNeighbors)
Set the number of neighbors that will be sampled in this layer.
|
Self |
setWeightInitScheme(WeightInitScheme weightInitScheme)
Set the weight initialization scheme for the weights of this layer.
|
Self |
useEdgeToEdgeConnection(boolean edgeToEdgeConnection)
Use the connection between edges and edges.
|
Self |
useEdgeToVertexConnection(boolean edgeToVertexConnection)
Use the connection between edges and vertices.
|
Self |
useVertexToEdgeConnection(boolean vertexToEdgeConnection)
Use the connection between vertices and edges.
|
Self |
useVertexToVertexConnection(boolean vertexToVertexConnection)
Use the connection between vertices and vertices.
|
public Self setActivationFunction(ActivationFunction activationFunction)
ActivationFunction
)
default: GraphWiseBaseConvLayerConfig.DEFAULT_ACTIVATION_FUNCTION
activationFunction
- activation functionpublic Self setDropoutRate(double dropoutRate)
dropoutRate
- dropout ratepublic Self setNumSampledNeighbors(int numSampledNeighbors)
default: GraphWiseBaseConvLayerConfig.DEFAULT_NUM_SAMPLED_NEIGHBORS
numSampledNeighbors
- number of neighbors to samplepublic Self setWeightInitScheme(WeightInitScheme weightInitScheme)
WeightInitScheme
.
default: GraphWiseBaseConvLayerConfig.DEFAULT_WEIGHT_INIT_SCHEME
weightInitScheme
- weight initialization schemepublic Self useEdgeToEdgeConnection(boolean edgeToEdgeConnection)
default: GraphWiseBaseConvLayerConfig#DEFAULT_EDGE_EDGE_CONNECTION
edgeToEdgeConnection
- boolean flag to use edge to edge connectionpublic Self useEdgeToVertexConnection(boolean edgeToVertexConnection)
default: GraphWiseBaseConvLayerConfig#DEFAULT_EDGE_VERTEX_CONNECTION
edgeToVertexConnection
- boolean flag to use edge to vertex connectionpublic Self useVertexToEdgeConnection(boolean vertexToEdgeConnection)
default: GraphWiseBaseConvLayerConfig#DEFAULT_VERTEX_EDGE_CONNECTION
vertexToEdgeConnection
- boolean flag to use vertex to edge connectionpublic Self useVertexToVertexConnection(boolean vertexToVertexConnection)
default: GraphWiseBaseConvLayerConfig#DEFAULT_VERTEX_VERTEX_CONNECTION
vertexToVertexConnection
- boolean flag to use vertex to vertex connection