Package org.neo4j.gds.ml.pipeline.linkPipeline

Interface LinkPredictionModelInfo

All Superinterfaces:

org.neo4j.gds.config.ToMapConvertible

public interface LinkPredictionModelInfo
extends org.neo4j.gds.config.ToMapConvertible

Method Summary

Modifier and Type	Method	Description
org.neo4j.gds.ml.models.TrainerConfig	bestParameters()
java.util.Map<java.lang.String,java.lang.Object>	metrics()
static LinkPredictionModelInfo	of(java.util.Map<org.neo4j.gds.ml.metrics.Metric,java.lang.Double> testMetrics, java.util.Map<org.neo4j.gds.ml.metrics.Metric,java.lang.Double> outerTrainMetrics, org.neo4j.gds.ml.metrics.ModelCandidateStats bestCandidate, LinkPredictionPredictPipeline pipeline)
LinkPredictionPredictPipeline	pipeline()
default java.util.Map<java.lang.String,java.lang.Object>	toMap()

Method Detail

bestParameters

org.neo4j.gds.ml.models.TrainerConfig bestParameters()

metrics

java.util.Map<java.lang.String,java.lang.Object> metrics()

pipeline

LinkPredictionPredictPipeline pipeline()

toMap

default java.util.Map<java.lang.String,java.lang.Object> toMap()

Specified by:

toMap in interface org.neo4j.gds.config.ToMapConvertible

of

static LinkPredictionModelInfo of(java.util.Map<org.neo4j.gds.ml.metrics.Metric,java.lang.Double> testMetrics,
                                  java.util.Map<org.neo4j.gds.ml.metrics.Metric,java.lang.Double> outerTrainMetrics,
                                  org.neo4j.gds.ml.metrics.ModelCandidateStats bestCandidate,
                                  LinkPredictionPredictPipeline pipeline)