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

Class LinkPredictionTrainPipelineExecutor

java.lang.Object

org.neo4j.gds.Algorithm<SELF,RESULT>

org.neo4j.gds.ml.pipeline.PipelineExecutor<LinkPredictionTrainConfig,LinkPredictionPipeline,org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionData,LinkPredictionTrainConfig,LinkPredictionModelInfo>,LinkPredictionTrainPipelineExecutor>

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

All Implemented Interfaces:

org.neo4j.gds.core.utils.TerminationFlag

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

Nested Class Summary

Nested classes/interfaces inherited from class org.neo4j.gds.ml.pipeline.PipelineExecutor

org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits, org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter

Field Summary

Fields inherited from class org.neo4j.gds.ml.pipeline.PipelineExecutor

caller, config, graphName, graphStore, pipeline

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
LinkPredictionTrainPipelineExecutor(LinkPredictionPipeline pipeline, LinkPredictionTrainConfig config, org.neo4j.gds.BaseProc caller, org.neo4j.gds.api.GraphStore graphStore, java.lang.String graphName, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

Method Summary

Modifier and Type	Method	Description
protected void	cleanUpGraphStore(java.util.Map<org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits,org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter> datasets)
protected org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionData,LinkPredictionTrainConfig,LinkPredictionModelInfo>	execute(java.util.Map<org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits,org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter> dataSplits)
LinkPredictionTrainPipelineExecutor	me()
java.util.Map<org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits,org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter>	splitDataset()

Methods inherited from class org.neo4j.gds.ml.pipeline.PipelineExecutor

compute, release, validate

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

Constructor Detail

LinkPredictionTrainPipelineExecutor

public LinkPredictionTrainPipelineExecutor(LinkPredictionPipeline pipeline,
                                           LinkPredictionTrainConfig config,
                                           org.neo4j.gds.BaseProc caller,
                                           org.neo4j.gds.api.GraphStore graphStore,
                                           java.lang.String graphName,
                                           org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

Method Detail

splitDataset

public java.util.Map<org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits,org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter> splitDataset()

Specified by:

splitDataset in class org.neo4j.gds.ml.pipeline.PipelineExecutor<LinkPredictionTrainConfig,LinkPredictionPipeline,org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionData,LinkPredictionTrainConfig,LinkPredictionModelInfo>,LinkPredictionTrainPipelineExecutor>

execute

protected org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionData,LinkPredictionTrainConfig,LinkPredictionModelInfo> execute(java.util.Map<org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits,org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter> dataSplits)

Specified by:

execute in class org.neo4j.gds.ml.pipeline.PipelineExecutor<LinkPredictionTrainConfig,LinkPredictionPipeline,org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionData,LinkPredictionTrainConfig,LinkPredictionModelInfo>,LinkPredictionTrainPipelineExecutor>

me

public LinkPredictionTrainPipelineExecutor me()

Specified by:

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

cleanUpGraphStore

protected void cleanUpGraphStore(java.util.Map<org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits,org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter> datasets)

Overrides:

cleanUpGraphStore in class org.neo4j.gds.ml.pipeline.PipelineExecutor<LinkPredictionTrainConfig,LinkPredictionPipeline,org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionData,LinkPredictionTrainConfig,LinkPredictionModelInfo>,LinkPredictionTrainPipelineExecutor>