Package org.neo4j.gds.ml.pipeline

Class TrainingPipeline<FEATURE_STEP extends FeatureStep>

java.lang.Object

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

All Implemented Interfaces:

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

Direct Known Subclasses:

LinkPredictionTrainingPipeline, NodePropertyTrainingPipeline

public abstract class TrainingPipeline<FEATURE_STEP extends FeatureStep>
extends java.lang.Object
implements Pipeline<FEATURE_STEP>

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
protected static class	TrainingPipeline.TrainingType

Field Summary

Fields

Modifier and Type	Field	Description
protected AutoTuningConfig	autoTuningConfig
protected java.util.List<FEATURE_STEP>	featureSteps
protected java.util.List<ExecutableNodePropertyStep>	nodePropertySteps
protected java.util.Map<org.neo4j.gds.ml.models.TrainingMethod,java.util.List<org.neo4j.gds.ml.models.automl.TunableTrainerConfig>>	trainingParameterSpace

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	TrainingPipeline(TrainingPipeline.TrainingType trainingType)

Method Summary

Modifier and Type	Method	Description
void	addFeatureStep(FEATURE_STEP featureStep)
protected abstract java.util.Map<java.lang.String,java.lang.Object>	additionalEntries()
void	addNodePropertyStep(ExecutableNodePropertyStep step)
void	addTrainerConfig(org.neo4j.gds.ml.models.automl.TunableTrainerConfig trainingConfig)
void	addTrainerConfig(org.neo4j.gds.ml.models.TrainerConfig trainingConfig)
AutoTuningConfig	autoTuningConfig()
java.time.ZonedDateTime	creationTime()
protected abstract java.util.Map<java.lang.String,java.util.List<java.util.Map<java.lang.String,java.lang.Object>>>	featurePipelineDescription()
java.util.List<FEATURE_STEP>	featureSteps()
java.util.List<ExecutableNodePropertyStep>	nodePropertySteps()
int	numberOfModelSelectionTrials()
void	setAutoTuningConfig(AutoTuningConfig autoTuningConfig)
java.util.Map<java.lang.String,java.lang.Object>	toMap()
static java.util.Map<java.lang.String,java.util.List<java.util.Map<java.lang.String,java.lang.Object>>>	toMapParameterSpace(java.util.Map<org.neo4j.gds.ml.models.TrainingMethod,java.util.List<org.neo4j.gds.ml.models.automl.TunableTrainerConfig>> parameterSpace)
java.util.Map<org.neo4j.gds.ml.models.TrainingMethod,java.util.List<org.neo4j.gds.ml.models.automl.TunableTrainerConfig>>	trainingParameterSpace()
abstract java.lang.String	type()
void	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

nodePropertySteps

protected final java.util.List<ExecutableNodePropertyStep> nodePropertySteps

featureSteps

protected final java.util.List<FEATURE_STEP extends FeatureStep> featureSteps

trainingParameterSpace

protected java.util.Map<org.neo4j.gds.ml.models.TrainingMethod,java.util.List<org.neo4j.gds.ml.models.automl.TunableTrainerConfig>> trainingParameterSpace

autoTuningConfig

protected AutoTuningConfig autoTuningConfig

Constructor Detail

TrainingPipeline

protected TrainingPipeline(TrainingPipeline.TrainingType trainingType)

Method Detail

toMapParameterSpace

public static java.util.Map<java.lang.String,java.util.List<java.util.Map<java.lang.String,java.lang.Object>>> toMapParameterSpace(java.util.Map<org.neo4j.gds.ml.models.TrainingMethod,java.util.List<org.neo4j.gds.ml.models.automl.TunableTrainerConfig>> parameterSpace)

toMap

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

Specified by:

toMap in interface org.neo4j.gds.config.ToMapConvertible

type

public abstract java.lang.String type()

featurePipelineDescription

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

additionalEntries

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

addNodePropertyStep

public void addNodePropertyStep(ExecutableNodePropertyStep step)

addFeatureStep

public void addFeatureStep(FEATURE_STEP featureStep)

nodePropertySteps

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

Specified by:

nodePropertySteps in interface Pipeline<FEATURE_STEP extends FeatureStep>

featureSteps

public java.util.List<FEATURE_STEP> featureSteps()

Specified by:

featureSteps in interface Pipeline<FEATURE_STEP extends FeatureStep>

trainingParameterSpace

public java.util.Map<org.neo4j.gds.ml.models.TrainingMethod,java.util.List<org.neo4j.gds.ml.models.automl.TunableTrainerConfig>> trainingParameterSpace()

numberOfModelSelectionTrials

public int numberOfModelSelectionTrials()

addTrainerConfig

public void addTrainerConfig(org.neo4j.gds.ml.models.automl.TunableTrainerConfig trainingConfig)

addTrainerConfig

public void addTrainerConfig(org.neo4j.gds.ml.models.TrainerConfig trainingConfig)

autoTuningConfig

public AutoTuningConfig autoTuningConfig()

setAutoTuningConfig

public void setAutoTuningConfig(AutoTuningConfig autoTuningConfig)

validateTrainingParameterSpace

public void validateTrainingParameterSpace()

creationTime

public java.time.ZonedDateTime creationTime()