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 NodePropertyPredictionSplitConfigsplitConfig-
Fields inherited from class org.neo4j.gds.ml.pipeline.TrainingPipeline
autoTuningConfig, featureSteps, nodePropertySteps, trainingParameterSpace
-
-
Constructor Summary
Constructors Modifier Constructor Description protectedNodePropertyTrainingPipeline(TrainingPipeline.TrainingType trainingType)
-
Method Summary
All Methods Instance Methods Abstract Methods Concrete Methods 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 booleanrequireEagerFeatures()voidsetSplitConfig(NodePropertyPredictionSplitConfig splitConfig)NodePropertyPredictionSplitConfigsplitConfig()-
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:
featurePipelineDescriptionin classTrainingPipeline<NodeFeatureStep>
-
additionalEntries
protected java.util.Map<java.lang.String,java.lang.Object> additionalEntries()
- Specified by:
additionalEntriesin classTrainingPipeline<NodeFeatureStep>
-
setSplitConfig
public void setSplitConfig(NodePropertyPredictionSplitConfig splitConfig)
-
splitConfig
public NodePropertyPredictionSplitConfig splitConfig()
-
requireEagerFeatures
public abstract boolean requireEagerFeatures()
-
-