Package org.neo4j.gds.embeddings.graphsage

Class GraphSageModelTrainer

java.lang.Object

org.neo4j.gds.embeddings.graphsage.GraphSageModelTrainer

public class GraphSageModelTrainer
extends java.lang.Object

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static interface	GraphSageModelTrainer.GraphSageTrainMetrics
static interface	GraphSageModelTrainer.ModelTrainResult

Constructor Summary

Constructors

Constructor	Description
GraphSageModelTrainer(GraphSageTrainConfig config, java.util.concurrent.ExecutorService executor, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)
GraphSageModelTrainer(GraphSageTrainConfig config, java.util.concurrent.ExecutorService executor, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker, FeatureFunction featureFunction, java.util.Collection<org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Matrix>> labelProjectionWeights)

Method Summary

Modifier and Type	Method	Description
static java.util.List<org.neo4j.gds.core.utils.progress.tasks.Task>	progressTasks(GraphSageTrainConfig config)
GraphSageModelTrainer.ModelTrainResult	train(org.neo4j.gds.api.Graph graph, org.neo4j.gds.core.utils.paged.HugeObjectArray<double[]> features)

Methods inherited from class java.lang.Object

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

Constructor Detail

GraphSageModelTrainer

public GraphSageModelTrainer(GraphSageTrainConfig config,
                             java.util.concurrent.ExecutorService executor,
                             org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

GraphSageModelTrainer

public GraphSageModelTrainer(GraphSageTrainConfig config,
                             java.util.concurrent.ExecutorService executor,
                             org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker,
                             FeatureFunction featureFunction,
                             java.util.Collection<org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Matrix>> labelProjectionWeights)

Method Detail

progressTasks

public static java.util.List<org.neo4j.gds.core.utils.progress.tasks.Task> progressTasks(GraphSageTrainConfig config)

train

public GraphSageModelTrainer.ModelTrainResult train(org.neo4j.gds.api.Graph graph,
                                                    org.neo4j.gds.core.utils.paged.HugeObjectArray<double[]> features)