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)

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

MeanAggregatingLayer

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

Method Detail

aggregator

public Aggregator aggregator()

Specified by:

aggregator in interface Layer

sampleSize

public int sampleSize()

Specified by:

sampleSize in interface Layer