Package org.neo4j.gds.embeddings.graphsage

Class MeanAggregatingLayer

java.lang.Object

org.neo4j.gds.embeddings.graphsage.MeanAggregatingLayer

All Implemented Interfaces:

Layer

public class MeanAggregatingLayer
extends java.lang.Object
implements Layer

Constructor Summary

Constructors

Constructor	Description
MeanAggregatingLayer(org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Matrix> weights, int sampleSize, ActivationFunction activationFunction, long randomSeed)

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

MeanAggregatingLayer

public MeanAggregatingLayer(org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Matrix> weights,
                            int sampleSize,
                            ActivationFunction activationFunction,
                            long randomSeed)

Method Detail

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

sampleSize

public int sampleSize()

Specified by:

sampleSize in interface Layer