Package org.neo4j.gds.embeddings.graphsage

Class MaxPoolAggregatingLayer

java.lang.Object

org.neo4j.gds.embeddings.graphsage.MaxPoolAggregatingLayer

All Implemented Interfaces:

Layer

public class MaxPoolAggregatingLayer
extends java.lang.Object
implements Layer

Constructor Summary

Constructors

Constructor

Description

MaxPoolAggregatingLayer(int sampleSize, org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Matrix> poolWeights, org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Matrix> selfWeights, org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Matrix> neighborsWeights, org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Vector> bias, ActivationFunction activationFunction)

Method Summary

Modifier and Type	Method	Description
Aggregator	aggregator()
int	sampleSize()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.neo4j.gds.embeddings.graphsage.Layer

weights

Constructor Detail

MaxPoolAggregatingLayer

public MaxPoolAggregatingLayer(int sampleSize,
                               org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Matrix> poolWeights,
                               org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Matrix> selfWeights,
                               org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Matrix> neighborsWeights,
                               org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Vector> bias,
                               ActivationFunction activationFunction)

Method Detail

sampleSize

public int sampleSize()

Specified by:

sampleSize in interface Layer

aggregator

public Aggregator aggregator()

Specified by:

aggregator in interface Layer