Package org.neo4j.gds.ml.training

Class TrainingStatistics

java.lang.Object
org.neo4j.gds.ml.training.TrainingStatistics
public final class TrainingStatistics extends Object

  • Constructor Details

    • TrainingStatistics

      public TrainingStatistics(List<? extends Metric> metrics)

  • Method Details

    • getTrainStats

      @TestOnly public List<EvaluationScores> getTrainStats(Metric metric)

    • getValidationStats

      @TestOnly public List<EvaluationScores> getValidationStats(Metric metric)

    • getTestScore

      @TestOnly public Double getTestScore(Metric metric)

    • toMap

      public Map<String,Object> toMap()
      Turns this class into a Cypher map, to be returned in a procedure YIELD field. This is intentionally omitting the test scores. These can be added to extend the return surface later.

    • getMainMetric

      public double getMainMetric(int trial)

    • validationMetricsAvg

      public Map<Metric,Double> validationMetricsAvg(int trial)

    • trainMetricsAvg

      public Map<Metric,Double> trainMetricsAvg(int trial)

    • evaluationMetric

      public Metric evaluationMetric()

    • addCandidateStats

      public void addCandidateStats(ModelCandidateStats statistics)

    • addTestScore

      public void addTestScore(Metric metric, double score)

    • addOuterTrainScore

      public void addOuterTrainScore(Metric metric, double score)

    • winningModelTestMetrics

      public Map<Metric,Double> winningModelTestMetrics()

    • winningModelOuterTrainMetrics

      public Map<Metric,Double> winningModelOuterTrainMetrics()

    • getBestTrialIdx

      public int getBestTrialIdx()

    • bestCandidate

      public ModelCandidateStats bestCandidate()

    • getBestTrialScore

      public double getBestTrialScore()

    • bestParameters

      public TrainerConfig bestParameters()

    • memoryEstimationStatsMap

      public static org.neo4j.gds.core.utils.mem.MemoryEstimation memoryEstimationStatsMap(int numberOfMetricsSpecifications, int numberOfModelCandidates)

    • memoryEstimationStatsMap

      public static org.neo4j.gds.core.utils.mem.MemoryEstimation memoryEstimationStatsMap(int numberOfMetricsSpecifications, int numberOfModelCandidates, int numberOfClasses)