Package org.neo4j.gds.ml.linkmodels
Class LinkPredictionTrainProc
- java.lang.Object
-
- org.neo4j.graphalgo.BaseProc
-
- org.neo4j.graphalgo.AlgoBaseProc<ALGO,org.neo4j.graphalgo.core.model.Model<TRAIN_RESULT,TRAIN_CONFIG>,TRAIN_CONFIG>
-
- org.neo4j.graphalgo.TrainProc<org.neo4j.gds.ml.linkmodels.LinkPredictionTrain,org.neo4j.gds.ml.linkmodels.logisticregression.LinkLogisticRegressionData,org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig>
-
- org.neo4j.gds.ml.linkmodels.LinkPredictionTrainProc
-
public class LinkPredictionTrainProc extends org.neo4j.graphalgo.TrainProc<org.neo4j.gds.ml.linkmodels.LinkPredictionTrain,org.neo4j.gds.ml.linkmodels.logisticregression.LinkLogisticRegressionData,org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig>
-
-
Nested Class Summary
-
Nested classes/interfaces inherited from class org.neo4j.graphalgo.TrainProc
org.neo4j.graphalgo.TrainProc.TrainResult
-
-
Constructor Summary
Constructors Constructor Description LinkPredictionTrainProc()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description protected org.neo4j.graphalgo.AlgorithmFactory<org.neo4j.gds.ml.linkmodels.LinkPredictionTrain,org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig>algorithmFactory()protected org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfignewConfig(java.lang.String username, java.util.Optional<java.lang.String> graphName, java.util.Optional<org.neo4j.graphalgo.config.GraphCreateConfig> maybeImplicitCreate, org.neo4j.graphalgo.core.CypherMapWrapper config)java.util.stream.Stream<MLTrainResult>train(java.lang.Object graphNameOrConfig, java.util.Map<java.lang.String,java.lang.Object> configuration)protected voidvalidateConfigs(org.neo4j.graphalgo.config.GraphCreateConfig graphCreateConfig, org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig config)-
Methods inherited from class org.neo4j.graphalgo.AlgoBaseProc
algoName, compute, compute, computeEstimate, createGraph, getOrCreateGraphStore, newConfig, nodeProperties, processInput, sharedConfigKeys, validateConfigsAndGraphStore, validateGraphStore
-
-
-
-
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.linkmodels.LinkPredictionTrainConfig newConfig(java.lang.String username, java.util.Optional<java.lang.String> graphName, java.util.Optional<org.neo4j.graphalgo.config.GraphCreateConfig> maybeImplicitCreate, org.neo4j.graphalgo.core.CypherMapWrapper config)- Specified by:
newConfigin classorg.neo4j.graphalgo.AlgoBaseProc<org.neo4j.gds.ml.linkmodels.LinkPredictionTrain,org.neo4j.graphalgo.core.model.Model<org.neo4j.gds.ml.linkmodels.logisticregression.LinkLogisticRegressionData,org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig>,org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig>
-
validateConfigs
protected void validateConfigs(org.neo4j.graphalgo.config.GraphCreateConfig graphCreateConfig, org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig config)- Overrides:
validateConfigsin classorg.neo4j.graphalgo.AlgoBaseProc<org.neo4j.gds.ml.linkmodels.LinkPredictionTrain,org.neo4j.graphalgo.core.model.Model<org.neo4j.gds.ml.linkmodels.logisticregression.LinkLogisticRegressionData,org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig>,org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig>
-
algorithmFactory
protected org.neo4j.graphalgo.AlgorithmFactory<org.neo4j.gds.ml.linkmodels.LinkPredictionTrain,org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig> algorithmFactory()
- Specified by:
algorithmFactoryin classorg.neo4j.graphalgo.AlgoBaseProc<org.neo4j.gds.ml.linkmodels.LinkPredictionTrain,org.neo4j.graphalgo.core.model.Model<org.neo4j.gds.ml.linkmodels.logisticregression.LinkLogisticRegressionData,org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig>,org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig>
-
-