Package org.neo4j.gds.ml.pipeline.nodePipeline.train

Class NodeClassificationTrain

java.lang.Object

org.neo4j.gds.ml.pipeline.nodePipeline.train.NodeClassificationTrain

public final class NodeClassificationTrain
extends java.lang.Object

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static interface	NodeClassificationTrain.ModelSelectResult

Method Summary

Modifier and Type	Method	Description
NodeClassificationTrainResult	compute()
static NodeClassificationTrain	create(org.neo4j.gds.api.Graph graph, NodeClassificationPipeline pipeline, NodeClassificationPipelineTrainConfig config, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)
static org.neo4j.gds.core.utils.mem.MemoryEstimation	estimate(NodeClassificationPipeline pipeline, NodeClassificationPipelineTrainConfig config)
static org.neo4j.gds.ml.core.subgraph.LocalIdMap	makeClassIdMap(org.neo4j.gds.core.utils.paged.HugeLongArray targets)
static org.neo4j.gds.core.utils.progress.tasks.Task	progressTask(int validationFolds, int paramsSize)
static java.lang.String	taskName()

Methods inherited from class java.lang.Object

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

Method Detail

estimate

public static org.neo4j.gds.core.utils.mem.MemoryEstimation estimate(NodeClassificationPipeline pipeline,
                                                                     NodeClassificationPipelineTrainConfig config)

taskName

public static java.lang.String taskName()

progressTask

public static org.neo4j.gds.core.utils.progress.tasks.Task progressTask(int validationFolds,
                                                                        int paramsSize)

create

public static NodeClassificationTrain create(org.neo4j.gds.api.Graph graph,
                                             NodeClassificationPipeline pipeline,
                                             NodeClassificationPipelineTrainConfig config,
                                             org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

makeClassIdMap

public static org.neo4j.gds.ml.core.subgraph.LocalIdMap makeClassIdMap(org.neo4j.gds.core.utils.paged.HugeLongArray targets)

compute

public NodeClassificationTrainResult compute()