Package org.neo4j.gds.ml.linkmodels.pipeline

Class LinkPredictionPipelineTrainProc

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<LinkPredictionTrain,org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionData,LinkPredictionTrainConfig,LinkPredictionModelInfo>

org.neo4j.gds.ml.linkmodels.pipeline.LinkPredictionPipelineTrainProc

public class LinkPredictionPipelineTrainProc
extends org.neo4j.gds.TrainProc<LinkPredictionTrain,org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionData,LinkPredictionTrainConfig,LinkPredictionModelInfo>

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 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

Constructor Summary

Constructors

Constructor	Description
LinkPredictionPipelineTrainProc()

Method Summary

Modifier and Type	Method	Description
protected org.neo4j.gds.AlgorithmFactory<LinkPredictionTrain,LinkPredictionTrainConfig>	algorithmFactory()
protected java.lang.String	modelType()
protected LinkPredictionTrainConfig	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> config)

Methods inherited from class org.neo4j.gds.TrainProc

trainAndStoreModelWithResult, validateConfigsBeforeLoad

Methods inherited from class org.neo4j.gds.AlgoBaseProc

algoName, compute, compute, computeEstimate, getOrCreateGraphStore, newConfig, nodeProperties, processInput, 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

Constructor Detail

LinkPredictionPipelineTrainProc

public LinkPredictionPipelineTrainProc()

Method Detail

train

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

newConfig

protected LinkPredictionTrainConfig 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<LinkPredictionTrain,org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionData,LinkPredictionTrainConfig,LinkPredictionModelInfo>,LinkPredictionTrainConfig>

algorithmFactory

protected org.neo4j.gds.AlgorithmFactory<LinkPredictionTrain,LinkPredictionTrainConfig> algorithmFactory()

Specified by:

algorithmFactory in class org.neo4j.gds.AlgoBaseProc<LinkPredictionTrain,org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionData,LinkPredictionTrainConfig,LinkPredictionModelInfo>,LinkPredictionTrainConfig>

modelType

protected java.lang.String modelType()

Specified by:

modelType in class org.neo4j.gds.TrainProc<LinkPredictionTrain,org.neo4j.gds.ml.linkmodels.pipeline.logisticRegression.LinkLogisticRegressionData,LinkPredictionTrainConfig,LinkPredictionModelInfo>