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

Class LinkPredictionTrain

java.lang.Object

org.neo4j.gds.Algorithm<LinkPredictionTrain,org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionData,LinkPredictionTrainConfig,LinkPredictionModelInfo>>

org.neo4j.gds.ml.linkmodels.pipeline.train.LinkPredictionTrain

All Implemented Interfaces:

org.neo4j.gds.core.utils.TerminationFlag

public class LinkPredictionTrain
extends org.neo4j.gds.Algorithm<LinkPredictionTrain,org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionData,LinkPredictionTrainConfig,LinkPredictionModelInfo>>

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static interface	LinkPredictionTrain.ModelSelectResult

Field Summary

Fields

Modifier and Type	Field	Description
static java.lang.String	MODEL_TYPE

Fields inherited from class org.neo4j.gds.Algorithm

progressTracker, terminationFlag

Fields inherited from interface org.neo4j.gds.core.utils.TerminationFlag

RUN_CHECK_NODE_COUNT, RUNNING_TRUE

Constructor Summary

Constructors

Constructor	Description
LinkPredictionTrain(org.neo4j.gds.api.GraphStore graphStore, LinkPredictionTrainConfig trainConfig, TrainingPipeline pipeline, PipelineExecutor pipelineExecutor, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

Method Summary

Modifier and Type	Method	Description
org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionData,LinkPredictionTrainConfig,LinkPredictionModelInfo>	compute()
org.neo4j.gds.core.utils.paged.HugeDoubleArray	extractTargets(long numberOfTargets, java.lang.String relationshipType)
LinkPredictionTrain	me()
void	release()

Methods inherited from class org.neo4j.gds.Algorithm

getProgressTracker, getTerminationFlag, running, withTerminationFlag

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.neo4j.gds.core.utils.TerminationFlag

assertRunning

Field Detail

MODEL_TYPE

public static final java.lang.String MODEL_TYPE

See Also:

Constant Field Values

Constructor Detail

LinkPredictionTrain

public LinkPredictionTrain(org.neo4j.gds.api.GraphStore graphStore,
                           LinkPredictionTrainConfig trainConfig,
                           TrainingPipeline pipeline,
                           PipelineExecutor pipelineExecutor,
                           org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

Method Detail

compute

public org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionData,LinkPredictionTrainConfig,LinkPredictionModelInfo> compute()

Specified by:

compute in class org.neo4j.gds.Algorithm<LinkPredictionTrain,org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionData,LinkPredictionTrainConfig,LinkPredictionModelInfo>>

extractTargets

public org.neo4j.gds.core.utils.paged.HugeDoubleArray extractTargets(long numberOfTargets,
                                                                     java.lang.String relationshipType)

me

public LinkPredictionTrain me()

Specified by:

me in class org.neo4j.gds.Algorithm<LinkPredictionTrain,org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionData,LinkPredictionTrainConfig,LinkPredictionModelInfo>>

release

public void release()

Specified by:

release in class org.neo4j.gds.Algorithm<LinkPredictionTrain,org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionData,LinkPredictionTrainConfig,LinkPredictionModelInfo>>