Package org.neo4j.gds.ml.pipeline

Class PipelineTrainAlgorithm<RESULT,MODEL_RESULT extends org.neo4j.gds.core.model.CatalogModelContainer<?,CONFIG,?>,CONFIG extends org.neo4j.gds.config.AlgoBaseConfig & org.neo4j.gds.model.ModelConfig,FEATURE_STEP extends FeatureStep>

java.lang.Object

org.neo4j.gds.Algorithm<MODEL_RESULT>

org.neo4j.gds.ml.pipeline.PipelineTrainAlgorithm<RESULT,MODEL_RESULT,CONFIG,FEATURE_STEP>

Direct Known Subclasses:

NodeClassificationTrainAlgorithm, NodeRegressionTrainAlgorithm

public abstract class PipelineTrainAlgorithm<RESULT,MODEL_RESULT extends org.neo4j.gds.core.model.CatalogModelContainer<?,CONFIG,?>,CONFIG extends org.neo4j.gds.config.AlgoBaseConfig & org.neo4j.gds.model.ModelConfig,FEATURE_STEP extends FeatureStep>
extends org.neo4j.gds.Algorithm<MODEL_RESULT>

Field Summary

Fields

Modifier and Type	Field	Description
protected CONFIG	config
protected org.neo4j.gds.api.GraphStore	graphStore
protected TrainingPipeline<FEATURE_STEP>	pipeline

Fields inherited from class org.neo4j.gds.Algorithm

progressTracker, terminationFlag

Constructor Summary

Constructors

Constructor	Description
PipelineTrainAlgorithm(PipelineTrainer<RESULT> pipelineTrainer, TrainingPipeline<FEATURE_STEP> pipeline, ResultToModelConverter<MODEL_RESULT,RESULT> toCatalogModelConverter, org.neo4j.gds.api.GraphStore graphStore, CONFIG config, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

Method Summary

Modifier and Type	Method	Description
MODEL_RESULT	compute()

Methods inherited from class org.neo4j.gds.Algorithm

getProgressTracker, getTerminationFlag, setTerminationFlag

Methods inherited from class java.lang.Object

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

Field Detail

pipeline

protected final TrainingPipeline<FEATURE_STEP extends FeatureStep> pipeline

graphStore

protected final org.neo4j.gds.api.GraphStore graphStore

config

protected final CONFIG extends org.neo4j.gds.config.AlgoBaseConfig & org.neo4j.gds.model.ModelConfig config

Constructor Detail

PipelineTrainAlgorithm

public PipelineTrainAlgorithm(PipelineTrainer<RESULT> pipelineTrainer,
                              TrainingPipeline<FEATURE_STEP> pipeline,
                              ResultToModelConverter<MODEL_RESULT,RESULT> toCatalogModelConverter,
                              org.neo4j.gds.api.GraphStore graphStore,
                              CONFIG config,
                              org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

Method Detail

compute

public MODEL_RESULT compute()

Specified by:

compute in class org.neo4j.gds.Algorithm<MODEL_RESULT extends org.neo4j.gds.core.model.CatalogModelContainer<?,CONFIG,?>>