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, long randomState)

Method Summary

Modifier and Type	Method	Description
Aggregator	aggregator()
org.neo4j.gds.ml.core.subgraph.NeighborhoodSampler	sampler()
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

neighborhoodFunction, randomState, 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,
                               long randomState)

Method Detail

sampleSize

public int sampleSize()

Specified by:

sampleSize in interface Layer

aggregator

public Aggregator aggregator()

Specified by:

aggregator in interface Layer

sampler

public org.neo4j.gds.ml.core.subgraph.NeighborhoodSampler sampler()

Specified by:

sampler in interface Layer