Package org.neo4j.gds.embeddings.graphsage

Class MeanAggregator

  • All Implemented Interfaces:
    Aggregator
    public class MeanAggregator
extends java.lang.Object
implements Aggregator

    • Constructor Summary

      Constructors 
      Constructor Description
      MeanAggregator​(org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Matrix> weights, ActivationFunction activationFunction)  

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      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)  
      Aggregator.AggregatorType type()  
      java.util.List<org.neo4j.gds.ml.core.functions.Weights<? extends org.neo4j.gds.ml.core.tensor.Tensor<?>>> weights()  
      org.neo4j.gds.ml.core.tensor.Matrix weightsData()  
      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

      • MeanAggregator

        public MeanAggregator​(org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Matrix> weights,
                      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

      • weightsData

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