Package org.neo4j.gds.ml.nodemodels.pipeline

Interface NodeClassificationPipelineModelInfo

  • All Superinterfaces:
    org.neo4j.gds.config.ToMapConvertible
    public interface NodeClassificationPipelineModelInfo
extends org.neo4j.gds.config.ToMapConvertible

    • Method Summary

      All Methods Static Methods Instance Methods Abstract Methods Default Methods 
      Modifier and Type Method Description
      org.neo4j.gds.ml.nodemodels.logisticregression.NodeLogisticRegressionTrainConfig bestParameters()
      The parameters that yielded the best fold-averaged validation score for the selection metric.
      static org.neo4j.gds.ml.nodemodels.pipeline.ImmutableNodeClassificationPipelineModelInfo.Builder builder()  
      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<org.neo4j.gds.ml.nodemodels.Metric,​org.neo4j.gds.ml.nodemodels.BestMetricData> metrics()  
      default java.util.Map<java.lang.String,​java.lang.Object> toMap()  
      NodeClassificationPipeline trainingPipeline()  

    • Method Detail

      • 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:

      • bestParameters

        org.neo4j.gds.ml.nodemodels.logisticregression.NodeLogisticRegressionTrainConfig bestParameters()
        The parameters that yielded the best fold-averaged validation score for the selection metric.
        Returns:

      • metrics

        java.util.Map<org.neo4j.gds.ml.nodemodels.Metric,​org.neo4j.gds.ml.nodemodels.BestMetricData> metrics()

      • toMap

        @Auxiliary
@Derived
default java.util.Map<java.lang.String,​java.lang.Object> toMap()
        Specified by:
        toMap in interface org.neo4j.gds.config.ToMapConvertible

      • builder

        static org.neo4j.gds.ml.nodemodels.pipeline.ImmutableNodeClassificationPipelineModelInfo.Builder builder()