Package org.neo4j.gds.ml.pipeline.linkPipeline

Class LinkPredictionTrainingPipeline

java.lang.Object

org.neo4j.gds.ml.pipeline.TrainingPipeline<LinkFeatureStep>

org.neo4j.gds.ml.pipeline.linkPipeline.LinkPredictionTrainingPipeline

All Implemented Interfaces:

org.neo4j.gds.config.ToMapConvertible, Pipeline<LinkFeatureStep>

public class LinkPredictionTrainingPipeline
extends TrainingPipeline<LinkFeatureStep>

Nested Class Summary

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

TrainingPipeline.TrainingType

Field Summary

Fields

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

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

autoTuningConfig, featureSteps, nodePropertySteps, trainingParameterSpace

Constructor Summary

Constructors

Constructor	Description
LinkPredictionTrainingPipeline()

Method Summary

Modifier and Type	Method	Description
protected java.util.Map<java.lang.String,java.lang.Object>	additionalEntries()
protected java.util.Map<java.lang.String,java.util.List<java.util.Map<java.lang.String,java.lang.Object>>>	featurePipelineDescription()
java.util.Optional<java.lang.String>	relationshipWeightProperty(org.neo4j.gds.executor.ExecutionContext executionContext)
void	setSplitConfig(LinkPredictionSplitConfig splitConfig)
void	specificValidateBeforeExecution(org.neo4j.gds.api.GraphStore graphStore)
LinkPredictionSplitConfig	splitConfig()
java.util.Map<java.lang.String,java.util.List<java.lang.String>>	tasksByRelationshipProperty(org.neo4j.gds.executor.ExecutionContext executionContext)
java.lang.String	type()

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

addFeatureStep, addNodePropertyStep, addTrainerConfig, addTrainerConfig, autoTuningConfig, creationTime, featureSteps, nodePropertySteps, numberOfModelSelectionTrials, setAutoTuningConfig, toMap, toMapParameterSpace, trainingParameterSpace, validateTrainingParameterSpace

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.ml.pipeline.Pipeline

featureProperties, featurePropertiesMissingFromGraph, validateBeforeExecution, validateFeatureProperties

Field Detail

PIPELINE_TYPE

public static final java.lang.String PIPELINE_TYPE

See Also:

Constant Field Values

MODEL_TYPE

public static final java.lang.String MODEL_TYPE

See Also:

Constant Field Values

Constructor Detail

LinkPredictionTrainingPipeline

public LinkPredictionTrainingPipeline()

Method Detail

type

public java.lang.String type()

Specified by:

type in class TrainingPipeline<LinkFeatureStep>

featurePipelineDescription

protected java.util.Map<java.lang.String,java.util.List<java.util.Map<java.lang.String,java.lang.Object>>> featurePipelineDescription()

Specified by:

featurePipelineDescription in class TrainingPipeline<LinkFeatureStep>

additionalEntries

protected java.util.Map<java.lang.String,java.lang.Object> additionalEntries()

Specified by:

additionalEntries in class TrainingPipeline<LinkFeatureStep>

splitConfig

public LinkPredictionSplitConfig splitConfig()

setSplitConfig

public void setSplitConfig(LinkPredictionSplitConfig splitConfig)

specificValidateBeforeExecution

public void specificValidateBeforeExecution(org.neo4j.gds.api.GraphStore graphStore)

tasksByRelationshipProperty

public java.util.Map<java.lang.String,java.util.List<java.lang.String>> tasksByRelationshipProperty(org.neo4j.gds.executor.ExecutionContext executionContext)

relationshipWeightProperty

public java.util.Optional<java.lang.String> relationshipWeightProperty(org.neo4j.gds.executor.ExecutionContext executionContext)