Package org.neo4j.gds

Class TrainProc<ALGO extends org.neo4j.gds.Algorithm<ALGO_RESULT>,ALGO_RESULT,TRAIN_CONFIG extends org.neo4j.gds.config.AlgoBaseConfig & org.neo4j.gds.model.ModelConfig,PROC_RESULT>

java.lang.Object

org.neo4j.gds.BaseProc

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

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

All Implemented Interfaces:

org.neo4j.gds.executor.AlgorithmSpec<ALGO,ALGO_RESULT,TRAIN_CONFIG,java.util.stream.Stream<PROC_RESULT>,org.neo4j.gds.AlgorithmFactory<?,ALGO,TRAIN_CONFIG>>

public abstract class TrainProc<ALGO extends org.neo4j.gds.Algorithm<ALGO_RESULT>,ALGO_RESULT,TRAIN_CONFIG extends org.neo4j.gds.config.AlgoBaseConfig & org.neo4j.gds.model.ModelConfig,PROC_RESULT>
extends AlgoBaseProc<ALGO,ALGO_RESULT,TRAIN_CONFIG,PROC_RESULT>

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	TrainProc.TrainingConfigValidation<TRAIN_CONFIG extends org.neo4j.gds.model.ModelConfig & org.neo4j.gds.config.AlgoBaseConfig>
static class	TrainProc.TrainResult

Field Summary

Fields inherited from class org.neo4j.gds.AlgoBaseProc

STATS_DESCRIPTION

Fields inherited from class org.neo4j.gds.BaseProc

api, callContext, ESTIMATE_DESCRIPTION, internalModelCatalog, log, procedureTransaction, taskRegistryFactory, transaction, userLogRegistryFactory, username

Constructor Summary

Constructors

Constructor	Description
TrainProc()

Method Summary

Modifier and Type	Method	Description
org.neo4j.gds.executor.ComputationResultConsumer<ALGO,ALGO_RESULT,TRAIN_CONFIG,java.util.stream.Stream<PROC_RESULT>>	computationResultConsumer()
protected abstract PROC_RESULT	constructProcResult(org.neo4j.gds.executor.ComputationResult<ALGO,ALGO_RESULT,TRAIN_CONFIG> computationResult)
protected abstract org.neo4j.gds.core.model.Model<?,?,?>	extractModel(ALGO_RESULT algo_result)
protected abstract java.lang.String	modelType()
protected java.util.stream.Stream<PROC_RESULT>	trainAndStoreModelWithResult(org.neo4j.gds.executor.ComputationResult<ALGO,ALGO_RESULT,TRAIN_CONFIG> computationResult)
org.neo4j.gds.executor.validation.ValidationConfiguration<TRAIN_CONFIG>	validationConfig()
org.neo4j.gds.executor.AlgorithmSpec<ALGO,ALGO_RESULT,TRAIN_CONFIG,java.util.stream.Stream<PROC_RESULT>,org.neo4j.gds.AlgorithmFactory<?,ALGO,TRAIN_CONFIG>>	withModelCatalog(org.neo4j.gds.core.model.ModelCatalog modelCatalog)

Methods inherited from class org.neo4j.gds.AlgoBaseProc

compute, compute, computeEstimate, configParser, name, newConfig, newConfigFunction, nodeProperties

Methods inherited from class org.neo4j.gds.BaseProc

databaseId, executionContext, graphLoaderContext, graphStoreFromCatalog, isGdsAdmin, memoryUsageValidator, modelCatalog, runWithExceptionLogging, runWithExceptionLogging, setModelCatalog, username, validateConfig, validateConfig, validateGraphName

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.neo4j.gds.executor.AlgorithmSpec

algorithmFactory, createDefaultExecutorSpec

Constructor Detail

TrainProc

public TrainProc()

Method Detail

modelType

protected abstract java.lang.String modelType()

constructProcResult

protected abstract PROC_RESULT constructProcResult(org.neo4j.gds.executor.ComputationResult<ALGO,ALGO_RESULT,TRAIN_CONFIG> computationResult)

extractModel

protected abstract org.neo4j.gds.core.model.Model<?,?,?> extractModel(ALGO_RESULT algo_result)

computationResultConsumer

public org.neo4j.gds.executor.ComputationResultConsumer<ALGO,ALGO_RESULT,TRAIN_CONFIG,java.util.stream.Stream<PROC_RESULT>> computationResultConsumer()

trainAndStoreModelWithResult

protected java.util.stream.Stream<PROC_RESULT> trainAndStoreModelWithResult(org.neo4j.gds.executor.ComputationResult<ALGO,ALGO_RESULT,TRAIN_CONFIG> computationResult)

validationConfig

public org.neo4j.gds.executor.validation.ValidationConfiguration<TRAIN_CONFIG> validationConfig()

Specified by:

validationConfig in interface org.neo4j.gds.executor.AlgorithmSpec<ALGO extends org.neo4j.gds.Algorithm<ALGO_RESULT>,ALGO_RESULT,TRAIN_CONFIG extends org.neo4j.gds.config.AlgoBaseConfig & org.neo4j.gds.model.ModelConfig,java.util.stream.Stream<PROC_RESULT>,org.neo4j.gds.AlgorithmFactory<?,ALGO extends org.neo4j.gds.Algorithm<ALGO_RESULT>,TRAIN_CONFIG extends org.neo4j.gds.config.AlgoBaseConfig & org.neo4j.gds.model.ModelConfig>>

Overrides:

validationConfig in class AlgoBaseProc<ALGO extends org.neo4j.gds.Algorithm<ALGO_RESULT>,ALGO_RESULT,TRAIN_CONFIG extends org.neo4j.gds.config.AlgoBaseConfig & org.neo4j.gds.model.ModelConfig,PROC_RESULT>

withModelCatalog

public org.neo4j.gds.executor.AlgorithmSpec<ALGO,ALGO_RESULT,TRAIN_CONFIG,java.util.stream.Stream<PROC_RESULT>,org.neo4j.gds.AlgorithmFactory<?,ALGO,TRAIN_CONFIG>> withModelCatalog(org.neo4j.gds.core.model.ModelCatalog modelCatalog)