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

Interface NodeClassificationTrain.ModelSelectResult

Enclosing class:

NodeClassificationTrain

public static interface NodeClassificationTrain.ModelSelectResult

Method Summary

Modifier and Type	Method	Description
org.neo4j.gds.models.TrainerConfig	bestParameters()
static NodeClassificationTrain.ModelSelectResult	of(org.neo4j.gds.models.TrainerConfig bestConfig, org.neo4j.gds.ml.nodemodels.StatsMap trainStats, org.neo4j.gds.ml.nodemodels.StatsMap validationStats)
default java.util.Map<java.lang.String,java.lang.Object>	toMap()
java.util.Map<org.neo4j.gds.ml.nodemodels.Metric,java.util.List<org.neo4j.gds.ml.nodemodels.ModelStats>>	trainStats()
java.util.Map<org.neo4j.gds.ml.nodemodels.Metric,java.util.List<org.neo4j.gds.ml.nodemodels.ModelStats>>	validationStats()

Method Detail

bestParameters

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

trainStats

java.util.Map<org.neo4j.gds.ml.nodemodels.Metric,java.util.List<org.neo4j.gds.ml.nodemodels.ModelStats>> trainStats()

validationStats

java.util.Map<org.neo4j.gds.ml.nodemodels.Metric,java.util.List<org.neo4j.gds.ml.nodemodels.ModelStats>> validationStats()

of

static NodeClassificationTrain.ModelSelectResult of(org.neo4j.gds.models.TrainerConfig bestConfig,
                                                    org.neo4j.gds.ml.nodemodels.StatsMap trainStats,
                                                    org.neo4j.gds.ml.nodemodels.StatsMap validationStats)

toMap

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