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.core.utils.mem.MemoryEstimation	embeddingsEstimation(GraphSageTrainConfig config, long batchSize, long nodeCount, int labelCount, boolean withGradientDescent)
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)

Methods inherited from class java.lang.Object

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

Method Detail

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)

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)