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

Class LinkPredictionPipeline

java.lang.Object

org.neo4j.gds.ml.pipeline.Pipeline<org.neo4j.gds.ml.linkmodels.pipeline.linkFeatures.LinkFeatureStep,org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionTrainConfig>

org.neo4j.gds.ml.linkmodels.pipeline.LinkPredictionPipeline

All Implemented Interfaces:

org.neo4j.gds.config.ToMapConvertible

public class LinkPredictionPipeline
extends org.neo4j.gds.ml.pipeline.Pipeline<org.neo4j.gds.ml.linkmodels.pipeline.linkFeatures.LinkFeatureStep,org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionTrainConfig>

Field Summary

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

featureSteps, nodePropertySteps, trainingParameterSpace

Constructor Summary

Constructors

Constructor	Description
LinkPredictionPipeline()

Method Summary

Modifier and Type	Method	Description
protected java.util.Map<java.lang.String,java.lang.Object>	additionalEntries()
LinkPredictionPipeline	copy()
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()
void	setSplitConfig(LinkPredictionSplitConfig splitConfig)
LinkPredictionSplitConfig	splitConfig()
java.util.Map<java.lang.String,java.util.List<java.lang.String>>	tasksByRelationshipProperty()
void	validate()

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

addFeatureStep, addNodePropertyStep, featureSteps, nodePropertySteps, setTrainingParameterSpace, toMap, trainingParameterSpace, validateBeforeExecution, validateFeatureProperties

Methods inherited from class java.lang.Object

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

Constructor Detail

LinkPredictionPipeline

public LinkPredictionPipeline()

Method Detail

copy

public LinkPredictionPipeline copy()

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 org.neo4j.gds.ml.pipeline.Pipeline<org.neo4j.gds.ml.linkmodels.pipeline.linkFeatures.LinkFeatureStep,org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionTrainConfig>

additionalEntries

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

Specified by:

additionalEntries in class org.neo4j.gds.ml.pipeline.Pipeline<org.neo4j.gds.ml.linkmodels.pipeline.linkFeatures.LinkFeatureStep,org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionTrainConfig>

splitConfig

public LinkPredictionSplitConfig splitConfig()

setSplitConfig

public void setSplitConfig(LinkPredictionSplitConfig splitConfig)

validate

public void validate()

tasksByRelationshipProperty

public java.util.Map<java.lang.String,java.util.List<java.lang.String>> tasksByRelationshipProperty()

relationshipWeightProperty

public java.util.Optional<java.lang.String> relationshipWeightProperty()