Package org.neo4j.gds.embeddings.graphsage

Class GraphSageHelper

java.lang.Object

org.neo4j.gds.embeddings.graphsage.GraphSageHelper

public final class GraphSageHelper
extends java.lang.Object

Method Summary

Modifier and Type	Method	Description
static org.neo4j.gds.ml.core.Variable<org.neo4j.gds.ml.core.tensor.Matrix>	embeddingsComputationGraph(java.util.List<org.neo4j.gds.ml.core.subgraph.SubGraph> subGraphs, Layer[] layers, org.neo4j.gds.ml.core.Variable<org.neo4j.gds.ml.core.tensor.Matrix> batchedFeaturesExtractor)
static org.neo4j.gds.ml.core.Variable<org.neo4j.gds.ml.core.tensor.Matrix>	embeddingsComputationGraph(org.neo4j.gds.api.Graph graph, boolean useWeights, long[] nodeIds, org.neo4j.gds.core.utils.paged.HugeObjectArray<double[]> features, Layer[] layers, FeatureFunction featureFunction)
static org.neo4j.gds.core.utils.mem.MemoryEstimation	embeddingsEstimation(GraphSageTrainConfig config, long batchSize, long nodeCount, int labelCount, boolean withGradientDescent)
static java.util.List<org.neo4j.gds.ml.core.features.FeatureExtractor>	featureExtractors(org.neo4j.gds.api.Graph graph, GraphSageTrainConfig config)
static org.neo4j.gds.core.utils.paged.HugeObjectArray<double[]>	initializeMultiLabelFeatures(org.neo4j.gds.api.Graph graph, MultiLabelFeatureExtractors multiLabelFeatureExtractors)
static org.neo4j.gds.core.utils.paged.HugeObjectArray<double[]>	initializeSingleLabelFeatures(org.neo4j.gds.api.Graph graph, GraphSageTrainConfig config)
static MultiLabelFeatureExtractors	multiLabelFeatureExtractors(org.neo4j.gds.api.Graph graph, GraphSageTrainConfig config)
static java.util.Map<org.neo4j.gds.NodeLabel,java.util.Set<java.lang.String>>	propertyKeysPerNodeLabel(org.neo4j.gds.api.schema.GraphSchema graphSchema)
static java.util.List<org.neo4j.gds.ml.core.subgraph.SubGraph>	subGraphsPerLayer(org.neo4j.gds.api.Graph graph, boolean useWeights, long[] nodeIds, Layer[] layers)

Methods inherited from class java.lang.Object

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

Method Detail

embeddingsComputationGraph

public static org.neo4j.gds.ml.core.Variable<org.neo4j.gds.ml.core.tensor.Matrix> embeddingsComputationGraph(org.neo4j.gds.api.Graph graph,
                                                                                                             boolean useWeights,
                                                                                                             long[] nodeIds,
                                                                                                             org.neo4j.gds.core.utils.paged.HugeObjectArray<double[]> features,
                                                                                                             Layer[] layers,
                                                                                                             FeatureFunction featureFunction)

embeddingsComputationGraph

public static org.neo4j.gds.ml.core.Variable<org.neo4j.gds.ml.core.tensor.Matrix> embeddingsComputationGraph(java.util.List<org.neo4j.gds.ml.core.subgraph.SubGraph> subGraphs,
                                                                                                             Layer[] layers,
                                                                                                             org.neo4j.gds.ml.core.Variable<org.neo4j.gds.ml.core.tensor.Matrix> batchedFeaturesExtractor)

subGraphsPerLayer

public static java.util.List<org.neo4j.gds.ml.core.subgraph.SubGraph> subGraphsPerLayer(org.neo4j.gds.api.Graph graph,
                                                                                        boolean useWeights,
                                                                                        long[] nodeIds,
                                                                                        Layer[] layers)

embeddingsEstimation

public static org.neo4j.gds.core.utils.mem.MemoryEstimation embeddingsEstimation(GraphSageTrainConfig config,
                                                                                 long batchSize,
                                                                                 long nodeCount,
                                                                                 int labelCount,
                                                                                 boolean withGradientDescent)

initializeSingleLabelFeatures

public static org.neo4j.gds.core.utils.paged.HugeObjectArray<double[]> initializeSingleLabelFeatures(org.neo4j.gds.api.Graph graph,
                                                                                                     GraphSageTrainConfig config)

featureExtractors

public static java.util.List<org.neo4j.gds.ml.core.features.FeatureExtractor> featureExtractors(org.neo4j.gds.api.Graph graph,
                                                                                                GraphSageTrainConfig config)

multiLabelFeatureExtractors

public static MultiLabelFeatureExtractors multiLabelFeatureExtractors(org.neo4j.gds.api.Graph graph,
                                                                      GraphSageTrainConfig config)

initializeMultiLabelFeatures

public static org.neo4j.gds.core.utils.paged.HugeObjectArray<double[]> initializeMultiLabelFeatures(org.neo4j.gds.api.Graph graph,
                                                                                                    MultiLabelFeatureExtractors multiLabelFeatureExtractors)

propertyKeysPerNodeLabel

public static java.util.Map<org.neo4j.gds.NodeLabel,java.util.Set<java.lang.String>> propertyKeysPerNodeLabel(org.neo4j.gds.api.schema.GraphSchema graphSchema)