Package org.neo4j.gds.embeddings.graphsage

Class MaxPoolingAggregator

java.lang.Object

org.neo4j.gds.embeddings.graphsage.MaxPoolingAggregator

All Implemented Interfaces:

Aggregator

public class MaxPoolingAggregator
extends java.lang.Object
implements Aggregator

Nested Class Summary

Nested classes/interfaces inherited from interface org.neo4j.gds.embeddings.graphsage.Aggregator

Aggregator.AggregatorType

Constructor Summary

Constructors

Constructor

Description

MaxPoolingAggregator(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
ActivationFunction	activationFunction()
org.neo4j.gds.ml.core.Variable<org.neo4j.gds.ml.core.tensor.Matrix>	aggregate(org.neo4j.gds.ml.core.Variable<org.neo4j.gds.ml.core.tensor.Matrix> previousLayerRepresentations, org.neo4j.gds.ml.core.subgraph.SubGraph subGraph)
org.neo4j.gds.ml.core.tensor.Vector	bias()
org.neo4j.gds.ml.core.tensor.Matrix	neighborsWeights()
org.neo4j.gds.ml.core.tensor.Matrix	poolWeights()
org.neo4j.gds.ml.core.tensor.Matrix	selfWeights()
Aggregator.AggregatorType	type()
java.util.List<org.neo4j.gds.ml.core.functions.Weights<? extends org.neo4j.gds.ml.core.tensor.Tensor<?>>>	weights()
java.util.List<org.neo4j.gds.ml.core.functions.Weights<? extends org.neo4j.gds.ml.core.tensor.Tensor<?>>>	weightsWithoutBias()

Methods inherited from class java.lang.Object

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

Constructor Detail

MaxPoolingAggregator

public MaxPoolingAggregator(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

aggregate

public org.neo4j.gds.ml.core.Variable<org.neo4j.gds.ml.core.tensor.Matrix> aggregate(org.neo4j.gds.ml.core.Variable<org.neo4j.gds.ml.core.tensor.Matrix> previousLayerRepresentations,
                                                                                     org.neo4j.gds.ml.core.subgraph.SubGraph subGraph)

Specified by:

aggregate in interface Aggregator

weights

public java.util.List<org.neo4j.gds.ml.core.functions.Weights<? extends org.neo4j.gds.ml.core.tensor.Tensor<?>>> weights()

Specified by:

weights in interface Aggregator

weightsWithoutBias

public java.util.List<org.neo4j.gds.ml.core.functions.Weights<? extends org.neo4j.gds.ml.core.tensor.Tensor<?>>> weightsWithoutBias()

Specified by:

weightsWithoutBias in interface Aggregator

type

public Aggregator.AggregatorType type()

Specified by:

type in interface Aggregator

activationFunction

public ActivationFunction activationFunction()

Specified by:

activationFunction in interface Aggregator

poolWeights

public org.neo4j.gds.ml.core.tensor.Matrix poolWeights()

selfWeights

public org.neo4j.gds.ml.core.tensor.Matrix selfWeights()

neighborsWeights

public org.neo4j.gds.ml.core.tensor.Matrix neighborsWeights()

bias

public org.neo4j.gds.ml.core.tensor.Vector bias()