Package org.neo4j.gds.ml.pipeline.nodePipeline

Class NodePropertyTrainingPipeline

java.lang.Object

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

org.neo4j.gds.ml.pipeline.nodePipeline.NodePropertyTrainingPipeline

All Implemented Interfaces:

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

Direct Known Subclasses:

NodeClassificationTrainingPipeline, NodeRegressionTrainingPipeline

public abstract class NodePropertyTrainingPipeline
extends TrainingPipeline<NodeFeatureStep>

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
protected NodePropertyPredictionSplitConfig	splitConfig

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

autoTuningConfig, featureSteps, nodePropertySteps, trainingParameterSpace

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	NodePropertyTrainingPipeline(TrainingPipeline.TrainingType trainingType)

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()
abstract boolean	requireEagerFeatures()
void	setSplitConfig(NodePropertyPredictionSplitConfig splitConfig)
NodePropertyPredictionSplitConfig	splitConfig()

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

addFeatureStep, addNodePropertyStep, addTrainerConfig, addTrainerConfig, autoTuningConfig, creationTime, featureSteps, nodePropertySteps, numberOfModelSelectionTrials, setAutoTuningConfig, toMap, toMapParameterSpace, trainingParameterSpace, type, 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, specificValidateBeforeExecution, validateBeforeExecution, validateFeatureProperties

Field Detail

splitConfig

protected NodePropertyPredictionSplitConfig splitConfig

Constructor Detail

NodePropertyTrainingPipeline

protected NodePropertyTrainingPipeline(TrainingPipeline.TrainingType trainingType)

Method Detail

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<NodeFeatureStep>

additionalEntries

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

Specified by:

additionalEntries in class TrainingPipeline<NodeFeatureStep>

setSplitConfig

public void setSplitConfig(NodePropertyPredictionSplitConfig splitConfig)

splitConfig

public NodePropertyPredictionSplitConfig splitConfig()

requireEagerFeatures

public abstract boolean requireEagerFeatures()