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