Package org.neo4j.gds.ml.nodemodels
Class NodeClassificationTrainPipelineAlgorithmFactory
- java.lang.Object
-
- org.neo4j.gds.AlgorithmFactory<NodeClassificationTrainPipelineExecutor,org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig>
-
- org.neo4j.gds.ml.nodemodels.NodeClassificationTrainPipelineAlgorithmFactory
-
public class NodeClassificationTrainPipelineAlgorithmFactory extends org.neo4j.gds.AlgorithmFactory<NodeClassificationTrainPipelineExecutor,org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig>
-
-
Constructor Summary
Constructors Constructor Description NodeClassificationTrainPipelineAlgorithmFactory(org.neo4j.gds.BaseProc caller, org.neo4j.kernel.database.NamedDatabaseId databaseId, org.neo4j.gds.core.model.ModelCatalog modelCatalog)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description protected NodeClassificationTrainPipelineExecutorbuild(org.neo4j.gds.api.Graph graph, org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig configuration, org.neo4j.gds.core.utils.mem.AllocationTracker allocationTracker, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)org.neo4j.gds.core.utils.mem.MemoryEstimationmemoryEstimation(org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig configuration)org.neo4j.gds.core.utils.progress.tasks.TaskprogressTask(org.neo4j.gds.api.Graph graph, org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig config)protected java.lang.StringtaskName()
-
-
-
Method Detail
-
build
protected NodeClassificationTrainPipelineExecutor build(org.neo4j.gds.api.Graph graph, org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig configuration, org.neo4j.gds.core.utils.mem.AllocationTracker allocationTracker, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)
- Specified by:
buildin classorg.neo4j.gds.AlgorithmFactory<NodeClassificationTrainPipelineExecutor,org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig>
-
memoryEstimation
public org.neo4j.gds.core.utils.mem.MemoryEstimation memoryEstimation(org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig configuration)
- Overrides:
memoryEstimationin classorg.neo4j.gds.AlgorithmFactory<NodeClassificationTrainPipelineExecutor,org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig>
-
taskName
protected java.lang.String taskName()
- Specified by:
taskNamein classorg.neo4j.gds.AlgorithmFactory<NodeClassificationTrainPipelineExecutor,org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig>
-
progressTask
public org.neo4j.gds.core.utils.progress.tasks.Task progressTask(org.neo4j.gds.api.Graph graph, org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig config)- Overrides:
progressTaskin classorg.neo4j.gds.AlgorithmFactory<NodeClassificationTrainPipelineExecutor,org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig>
-
-