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

Interface NodeClassificationPipelineModelInfo

All Superinterfaces:

org.neo4j.gds.config.ToMapConvertible

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

Method Summary

Modifier and Type	Method	Description
org.neo4j.gds.ml.models.TrainerConfig	bestParameters()
java.util.List<java.lang.Long>	classes()	The distinct values of the target property which represent the allowed classes that the model can predict.
java.util.Map<java.lang.String,java.lang.Object>	metrics()
static NodeClassificationPipelineModelInfo	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, NodePropertyPredictPipeline pipeline, java.util.List<java.lang.Long> classes)
NodePropertyPredictPipeline	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

NodePropertyPredictPipeline pipeline()

classes

java.util.List<java.lang.Long> classes()

The distinct values of the target property which represent the allowed classes that the model can predict.

Returns:

toMap

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

Specified by:

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

of

static NodeClassificationPipelineModelInfo 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,
                                              NodePropertyPredictPipeline pipeline,
                                              java.util.List<java.lang.Long> classes)