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

Class TrainingPipeline

java.lang.Object

org.neo4j.gds.ml.linkmodels.pipeline.TrainingPipeline

All Implemented Interfaces:

org.neo4j.gds.core.model.Model.Mappable

public class TrainingPipeline
extends java.lang.Object
implements org.neo4j.gds.core.model.Model.Mappable

Constructor Summary

Constructors

Constructor	Description
TrainingPipeline()

Method Summary

Modifier and Type	Method	Description
void	addFeatureStep(org.neo4j.gds.ml.linkmodels.pipeline.linkFeatures.LinkFeatureStep step)
void	addNodePropertyStep(NodePropertyStep step)
TrainingPipeline	copy()
java.util.List<org.neo4j.gds.ml.linkmodels.pipeline.linkFeatures.LinkFeatureStep>	featureSteps()
java.util.List<NodePropertyStep>	nodePropertySteps()
java.util.List<org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionTrainConfig>	parameterConfigs(int concurrency)
java.util.List<java.util.Map<java.lang.String,java.lang.Object>>	parameterSpace()
java.util.Optional<java.lang.String>	relationshipWeightProperty()
void	setParameterSpace(@NotNull java.util.List<java.util.Map<java.lang.String,java.lang.Object>> parameterList)
void	setSplitConfig(@NotNull LinkPredictionSplitConfig splitConfig)
LinkPredictionSplitConfig	splitConfig()
java.util.Map<java.lang.String,java.util.List<java.lang.String>>	tasksByRelationshipProperty()
java.util.Map<java.lang.String,java.lang.Object>	toMap()
void	validate()
void	validate(org.neo4j.gds.api.Graph graph)

Methods inherited from class java.lang.Object

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

Constructor Detail

TrainingPipeline

public TrainingPipeline()

Method Detail

copy

public TrainingPipeline copy()

toMap

public java.util.Map<java.lang.String,java.lang.Object> toMap()

Specified by:

toMap in interface org.neo4j.gds.core.model.Model.Mappable

nodePropertySteps

public java.util.List<NodePropertyStep> nodePropertySteps()

addNodePropertyStep

public void addNodePropertyStep(NodePropertyStep step)

featureSteps

public java.util.List<org.neo4j.gds.ml.linkmodels.pipeline.linkFeatures.LinkFeatureStep> featureSteps()

addFeatureStep

public void addFeatureStep(org.neo4j.gds.ml.linkmodels.pipeline.linkFeatures.LinkFeatureStep step)

splitConfig

public LinkPredictionSplitConfig splitConfig()

setSplitConfig

public void setSplitConfig(@NotNull
                           @NotNull LinkPredictionSplitConfig splitConfig)

parameterSpace

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

setParameterSpace

public void setParameterSpace(@NotNull
                              @NotNull java.util.List<java.util.Map<java.lang.String,java.lang.Object>> parameterList)

parameterConfigs

public java.util.List<org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionTrainConfig> parameterConfigs(int concurrency)

validate

public void validate(org.neo4j.gds.api.Graph graph)

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()