Package org.neo4j.gds.ml.nodemodels.pipeline.predict

Class NodeClassificationPipelineTrainProc

java.lang.Object

org.neo4j.gds.BaseProc

org.neo4j.gds.AlgoBaseProc<ALGO,org.neo4j.gds.core.model.Model<TRAIN_RESULT,TRAIN_CONFIG,TRAIN_INFO>,TRAIN_CONFIG>

org.neo4j.gds.TrainProc<NodeClassificationTrainPipelineExecutor,org.neo4j.gds.ml.nodemodels.logisticregression.NodeLogisticRegressionData,org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig,NodeClassificationPipelineModelInfo>

org.neo4j.gds.ml.nodemodels.pipeline.predict.NodeClassificationPipelineTrainProc

public class NodeClassificationPipelineTrainProc
extends org.neo4j.gds.TrainProc<NodeClassificationTrainPipelineExecutor,org.neo4j.gds.ml.nodemodels.logisticregression.NodeLogisticRegressionData,org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig,NodeClassificationPipelineModelInfo>

Nested Class Summary

Nested classes/interfaces inherited from class org.neo4j.gds.TrainProc

org.neo4j.gds.TrainProc.TrainResult

Nested classes/interfaces inherited from class org.neo4j.gds.AlgoBaseProc

org.neo4j.gds.AlgoBaseProc.ComputationResult<A extends org.neo4j.gds.Algorithm<A,RESULT>,RESULT extends java.lang.Object,CONFIG extends org.neo4j.gds.config.AlgoBaseConfig>

Nested classes/interfaces inherited from class org.neo4j.gds.BaseProc

org.neo4j.gds.BaseProc.FreeMemoryInspector

Field Summary

Fields

Modifier and Type	Field	Description
org.neo4j.gds.core.model.ModelCatalog	modelCatalog

Fields inherited from class org.neo4j.gds.AlgoBaseProc

STATS_DESCRIPTION

Fields inherited from class org.neo4j.gds.BaseProc

allocationTracker, api, callContext, ESTIMATE_DESCRIPTION, log, procedureTransaction, taskRegistryFactory, transaction, username

Constructor Summary

Constructors

Constructor	Description
NodeClassificationPipelineTrainProc()

Method Summary

Modifier and Type	Method	Description
protected org.neo4j.gds.AlgorithmFactory<NodeClassificationTrainPipelineExecutor,org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig>	algorithmFactory()
protected java.lang.String	modelType()
protected org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig	newConfig(java.lang.String username, java.util.Optional<java.lang.String> graphName, java.util.Optional<org.neo4j.gds.config.GraphCreateConfig> maybeImplicitCreate, org.neo4j.gds.core.CypherMapWrapper config)
java.util.stream.Stream<MLTrainResult>	train(java.lang.Object graphNameOrConfig, java.util.Map<java.lang.String,java.lang.Object> configuration)

Methods inherited from class org.neo4j.gds.TrainProc

trainAndStoreModelWithResult, validateConfigsBeforeLoad

Methods inherited from class org.neo4j.gds.AlgoBaseProc

compute, compute, computeEstimate, getOrCreateGraphStore, newConfig, nodeProperties, processInput, procName, sharedConfigKeys, validateConfigsAfterLoad, validateConfigWithGraphStore

Methods inherited from class org.neo4j.gds.BaseProc

allocationTracker, catalogRequest, databaseId, estimateGraphCreate, graphStoreFromCatalog, isGdsAdmin, newLoader, runWithExceptionLogging, runWithExceptionLogging, tryValidateMemoryUsage, tryValidateMemoryUsage, username, validateConfig, validateConfig, validateGraphName

Methods inherited from class java.lang.Object

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

Field Detail

modelCatalog

public org.neo4j.gds.core.model.ModelCatalog modelCatalog

Constructor Detail

NodeClassificationPipelineTrainProc

public NodeClassificationPipelineTrainProc()

Method Detail

train

public java.util.stream.Stream<MLTrainResult> train(java.lang.Object graphNameOrConfig,
                                                    java.util.Map<java.lang.String,java.lang.Object> configuration)

newConfig

protected org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig newConfig(java.lang.String username,
                                                                                               java.util.Optional<java.lang.String> graphName,
                                                                                               java.util.Optional<org.neo4j.gds.config.GraphCreateConfig> maybeImplicitCreate,
                                                                                               org.neo4j.gds.core.CypherMapWrapper config)

Specified by:

newConfig in class org.neo4j.gds.AlgoBaseProc<NodeClassificationTrainPipelineExecutor,org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.nodemodels.logisticregression.NodeLogisticRegressionData,org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig,NodeClassificationPipelineModelInfo>,org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig>

algorithmFactory

protected org.neo4j.gds.AlgorithmFactory<NodeClassificationTrainPipelineExecutor,org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig> algorithmFactory()

Specified by:

algorithmFactory in class org.neo4j.gds.AlgoBaseProc<NodeClassificationTrainPipelineExecutor,org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.nodemodels.logisticregression.NodeLogisticRegressionData,org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig,NodeClassificationPipelineModelInfo>,org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig>

modelType

protected java.lang.String modelType()

Specified by:

modelType in class org.neo4j.gds.TrainProc<NodeClassificationTrainPipelineExecutor,org.neo4j.gds.ml.nodemodels.logisticregression.NodeLogisticRegressionData,org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig,NodeClassificationPipelineModelInfo>