Class NodeClassificationPipelineTrainProc
- java.lang.Object
-
- org.neo4j.gds.BaseProc
-
- org.neo4j.gds.AlgoBaseProc<ALGO,ALGO_RESULT,TRAIN_CONFIG,PROC_RESULT>
-
- org.neo4j.gds.TrainProc<NodeClassificationTrainPipelineExecutor,NodeClassificationPipelineTrainResult,org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfig,NodeClassificationPipelineTrainProc.NCTrainResult>
-
- org.neo4j.gds.ml.nodemodels.pipeline.predict.NodeClassificationPipelineTrainProc
-
- All Implemented Interfaces:
org.neo4j.gds.executor.AlgorithmSpec<NodeClassificationTrainPipelineExecutor,NodeClassificationPipelineTrainResult,org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfig,java.util.stream.Stream<NodeClassificationPipelineTrainProc.NCTrainResult>,org.neo4j.gds.AlgorithmFactory<?,NodeClassificationTrainPipelineExecutor,org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfig>>
public class NodeClassificationPipelineTrainProc extends org.neo4j.gds.TrainProc<NodeClassificationTrainPipelineExecutor,NodeClassificationPipelineTrainResult,org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfig,NodeClassificationPipelineTrainProc.NCTrainResult>
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static classNodeClassificationPipelineTrainProc.NCTrainResult
-
Constructor Summary
Constructors Constructor Description NodeClassificationPipelineTrainProc()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description org.neo4j.gds.GraphStoreAlgorithmFactory<NodeClassificationTrainPipelineExecutor,org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfig>algorithmFactory()protected NodeClassificationPipelineTrainProc.NCTrainResultconstructProcResult(org.neo4j.gds.executor.ComputationResult<NodeClassificationTrainPipelineExecutor,NodeClassificationPipelineTrainResult,org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfig> computationResult)java.util.stream.Stream<org.neo4j.gds.results.MemoryEstimateResult>estimate(java.lang.Object graphNameOrConfiguration, java.util.Map<java.lang.String,java.lang.Object> algoConfiguration)protected org.neo4j.gds.core.model.Model<?,?,?>extractModel(NodeClassificationPipelineTrainResult algoResult)protected java.lang.StringmodelType()protected org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfignewConfig(java.lang.String username, org.neo4j.gds.core.CypherMapWrapper config)java.util.stream.Stream<NodeClassificationPipelineTrainProc.NCTrainResult>train(java.lang.String graphName, java.util.Map<java.lang.String,java.lang.Object> configuration)-
Methods inherited from class org.neo4j.gds.TrainProc
computationResultConsumer, trainAndStoreModelWithResult, validationConfig
-
Methods inherited from class org.neo4j.gds.AlgoBaseProc
compute, compute, computeEstimate, configParser, name, newConfigFunction, nodeProperties, validator
-
-
-
-
Method Detail
-
train
public java.util.stream.Stream<NodeClassificationPipelineTrainProc.NCTrainResult> train(java.lang.String graphName, java.util.Map<java.lang.String,java.lang.Object> configuration)
-
estimate
public java.util.stream.Stream<org.neo4j.gds.results.MemoryEstimateResult> estimate(java.lang.Object graphNameOrConfiguration, java.util.Map<java.lang.String,java.lang.Object> algoConfiguration)
-
newConfig
protected org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfig newConfig(java.lang.String username, org.neo4j.gds.core.CypherMapWrapper config)- Specified by:
newConfigin classorg.neo4j.gds.AlgoBaseProc<NodeClassificationTrainPipelineExecutor,NodeClassificationPipelineTrainResult,org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfig,NodeClassificationPipelineTrainProc.NCTrainResult>
-
algorithmFactory
public org.neo4j.gds.GraphStoreAlgorithmFactory<NodeClassificationTrainPipelineExecutor,org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfig> algorithmFactory()
- Specified by:
algorithmFactoryin interfaceorg.neo4j.gds.executor.AlgorithmSpec<NodeClassificationTrainPipelineExecutor,NodeClassificationPipelineTrainResult,org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfig,java.util.stream.Stream<NodeClassificationPipelineTrainProc.NCTrainResult>,org.neo4j.gds.AlgorithmFactory<?,NodeClassificationTrainPipelineExecutor,org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfig>>- Specified by:
algorithmFactoryin classorg.neo4j.gds.AlgoBaseProc<NodeClassificationTrainPipelineExecutor,NodeClassificationPipelineTrainResult,org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfig,NodeClassificationPipelineTrainProc.NCTrainResult>
-
modelType
protected java.lang.String modelType()
- Specified by:
modelTypein classorg.neo4j.gds.TrainProc<NodeClassificationTrainPipelineExecutor,NodeClassificationPipelineTrainResult,org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfig,NodeClassificationPipelineTrainProc.NCTrainResult>
-
constructProcResult
protected NodeClassificationPipelineTrainProc.NCTrainResult constructProcResult(org.neo4j.gds.executor.ComputationResult<NodeClassificationTrainPipelineExecutor,NodeClassificationPipelineTrainResult,org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfig> computationResult)
- Specified by:
constructProcResultin classorg.neo4j.gds.TrainProc<NodeClassificationTrainPipelineExecutor,NodeClassificationPipelineTrainResult,org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfig,NodeClassificationPipelineTrainProc.NCTrainResult>
-
extractModel
protected org.neo4j.gds.core.model.Model<?,?,?> extractModel(NodeClassificationPipelineTrainResult algoResult)
- Specified by:
extractModelin classorg.neo4j.gds.TrainProc<NodeClassificationTrainPipelineExecutor,NodeClassificationPipelineTrainResult,org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfig,NodeClassificationPipelineTrainProc.NCTrainResult>
-
-