Package org.neo4j.gds.ml.training

Class CrossValidation<MODEL_TYPE>

java.lang.Object

org.neo4j.gds.ml.training.CrossValidation<MODEL_TYPE>

public class CrossValidation<MODEL_TYPE>
extends Object

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static interface	CrossValidation.ModelEvaluator<MODEL_TYPE>
static interface	CrossValidation.ModelTrainer<MODEL_TYPE>

Constructor Summary

Constructors

Constructor	Description
CrossValidation(org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker, org.neo4j.gds.termination.TerminationFlag terminationFlag, List<? extends Metric> metrics, int validationFolds, Optional<Long> randomSeed, CrossValidation.ModelTrainer<MODEL_TYPE> modelTrainer, CrossValidation.ModelEvaluator<MODEL_TYPE> modelEvaluator)

Method Summary

Modifier and Type	Method	Description
static List<org.neo4j.gds.core.utils.progress.tasks.Task>	progressTasks(int validationFolds, int numberOfModelSelectionTrials, long trainSetSize)
void	selectModel(org.neo4j.gds.core.utils.paged.ReadOnlyHugeLongArray outerTrainSet, org.eclipse.collections.api.block.function.primitive.LongToLongFunction targets, SortedSet<Long> distinctInternalTargets, TrainingStatistics trainingStatistics, Iterator<TrainerConfig> modelCandidates)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

CrossValidation

public CrossValidation(org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker,
 org.neo4j.gds.termination.TerminationFlag terminationFlag,
 List<? extends Metric> metrics,
 int validationFolds,
 Optional<Long> randomSeed,
 CrossValidation.ModelTrainer<MODEL_TYPE> modelTrainer,
 CrossValidation.ModelEvaluator<MODEL_TYPE> modelEvaluator)

Method Details

progressTasks

public static List<org.neo4j.gds.core.utils.progress.tasks.Task> progressTasks(int validationFolds,
 int numberOfModelSelectionTrials,
 long trainSetSize)

selectModel

public void selectModel(org.neo4j.gds.core.utils.paged.ReadOnlyHugeLongArray outerTrainSet,
 org.eclipse.collections.api.block.function.primitive.LongToLongFunction targets,
 SortedSet<Long> distinctInternalTargets,
 TrainingStatistics trainingStatistics,
 Iterator<TrainerConfig> modelCandidates)