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

Class LinkPredictionPipelineTrainProc

java.lang.Object

org.neo4j.gds.BaseProc

org.neo4j.gds.AlgoBaseProc<ALGO,ALGO_RESULT,TRAIN_CONFIG,PROC_RESULT>

org.neo4j.gds.TrainProc<LinkPredictionTrainPipelineExecutor,LinkPredictionTrainResult,LinkPredictionTrainConfig,LinkPredictionPipelineTrainProc.LPTrainResult>

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

All Implemented Interfaces:

org.neo4j.gds.executor.AlgorithmSpec<LinkPredictionTrainPipelineExecutor,LinkPredictionTrainResult,LinkPredictionTrainConfig,java.util.stream.Stream<LinkPredictionPipelineTrainProc.LPTrainResult>,org.neo4j.gds.AlgorithmFactory<?,LinkPredictionTrainPipelineExecutor,LinkPredictionTrainConfig>>

public class LinkPredictionPipelineTrainProc
extends org.neo4j.gds.TrainProc<LinkPredictionTrainPipelineExecutor,LinkPredictionTrainResult,LinkPredictionTrainConfig,LinkPredictionPipelineTrainProc.LPTrainResult>

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	LinkPredictionPipelineTrainProc.LPTrainResult

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

org.neo4j.gds.TrainProc.TrainingConfigValidation<TRAIN_CONFIG extends org.neo4j.gds.model.ModelConfig & org.neo4j.gds.config.AlgoBaseConfig>, org.neo4j.gds.TrainProc.TrainResult

Field Summary

Fields inherited from class org.neo4j.gds.TrainProc

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, userLogRegistryFactory, username

Constructor Summary

Constructors

Constructor	Description
LinkPredictionPipelineTrainProc()

Method Summary

Modifier and Type	Method	Description
org.neo4j.gds.GraphStoreAlgorithmFactory<LinkPredictionTrainPipelineExecutor,LinkPredictionTrainConfig>	algorithmFactory()
protected LinkPredictionPipelineTrainProc.LPTrainResult	constructProcResult(org.neo4j.gds.executor.ComputationResult<LinkPredictionTrainPipelineExecutor,LinkPredictionTrainResult,LinkPredictionTrainConfig> computationResult)
protected org.neo4j.gds.core.model.Model<?,?,?>	extractModel(LinkPredictionTrainResult algoResult)
protected java.lang.String	modelType()
protected LinkPredictionTrainConfig	newConfig(java.lang.String username, org.neo4j.gds.core.CypherMapWrapper config)
java.util.stream.Stream<LinkPredictionPipelineTrainProc.LPTrainResult>	train(java.lang.String graphName, java.util.Map<java.lang.String,java.lang.Object> config)

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

Methods inherited from class org.neo4j.gds.BaseProc

allocationTracker, databaseId, executionContext, graphLoaderContext, graphStoreFromCatalog, isGdsAdmin, memoryUsageValidator, runWithExceptionLogging, runWithExceptionLogging, 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<LinkPredictionPipelineTrainProc.LPTrainResult> train(java.lang.String graphName,
                                                                                    java.util.Map<java.lang.String,java.lang.Object> config)

newConfig

protected LinkPredictionTrainConfig newConfig(java.lang.String username,
                                              org.neo4j.gds.core.CypherMapWrapper config)

Specified by:

newConfig in class org.neo4j.gds.AlgoBaseProc<LinkPredictionTrainPipelineExecutor,LinkPredictionTrainResult,LinkPredictionTrainConfig,LinkPredictionPipelineTrainProc.LPTrainResult>

algorithmFactory

public org.neo4j.gds.GraphStoreAlgorithmFactory<LinkPredictionTrainPipelineExecutor,LinkPredictionTrainConfig> algorithmFactory()

Specified by:

algorithmFactory in interface org.neo4j.gds.executor.AlgorithmSpec<LinkPredictionTrainPipelineExecutor,LinkPredictionTrainResult,LinkPredictionTrainConfig,java.util.stream.Stream<LinkPredictionPipelineTrainProc.LPTrainResult>,org.neo4j.gds.AlgorithmFactory<?,LinkPredictionTrainPipelineExecutor,LinkPredictionTrainConfig>>

Specified by:

algorithmFactory in class org.neo4j.gds.AlgoBaseProc<LinkPredictionTrainPipelineExecutor,LinkPredictionTrainResult,LinkPredictionTrainConfig,LinkPredictionPipelineTrainProc.LPTrainResult>

modelType

protected java.lang.String modelType()

Specified by:

modelType in class org.neo4j.gds.TrainProc<LinkPredictionTrainPipelineExecutor,LinkPredictionTrainResult,LinkPredictionTrainConfig,LinkPredictionPipelineTrainProc.LPTrainResult>

extractModel

protected org.neo4j.gds.core.model.Model<?,?,?> extractModel(LinkPredictionTrainResult algoResult)

Specified by:

extractModel in class org.neo4j.gds.TrainProc<LinkPredictionTrainPipelineExecutor,LinkPredictionTrainResult,LinkPredictionTrainConfig,LinkPredictionPipelineTrainProc.LPTrainResult>

constructProcResult

protected LinkPredictionPipelineTrainProc.LPTrainResult constructProcResult(org.neo4j.gds.executor.ComputationResult<LinkPredictionTrainPipelineExecutor,LinkPredictionTrainResult,LinkPredictionTrainConfig> computationResult)

Specified by:

constructProcResult in class org.neo4j.gds.TrainProc<LinkPredictionTrainPipelineExecutor,LinkPredictionTrainResult,LinkPredictionTrainConfig,LinkPredictionPipelineTrainProc.LPTrainResult>