Package org.neo4j.gds.ml.models.mlp

Class MLPClassifierTrainer

java.lang.Object

org.neo4j.gds.ml.models.mlp.MLPClassifierTrainer

All Implemented Interfaces:

ClassifierTrainer

public class MLPClassifierTrainer
extends Object
implements ClassifierTrainer

Constructor Summary

Constructors

Constructor	Description
MLPClassifierTrainer(int numberOfClasses, MLPClassifierTrainConfig trainConfig, Optional<Long> randomSeed, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker, org.neo4j.gds.core.utils.progress.tasks.LogLevel messageLogLevel, org.neo4j.gds.termination.TerminationFlag terminationFlag, int concurrency)

Method Summary

Modifier and Type	Method	Description
MLPClassifier	train(Features features, org.neo4j.gds.collections.ha.HugeIntArray labels, org.neo4j.gds.core.utils.paged.ReadOnlyHugeLongArray trainSet)

Methods inherited from class java.lang.Object

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

Constructor Details

MLPClassifierTrainer

public MLPClassifierTrainer(int numberOfClasses,
 MLPClassifierTrainConfig trainConfig,
 Optional<Long> randomSeed,
 org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker,
 org.neo4j.gds.core.utils.progress.tasks.LogLevel messageLogLevel,
 org.neo4j.gds.termination.TerminationFlag terminationFlag,
 int concurrency)

Method Details

train

public MLPClassifier train(Features features,
 org.neo4j.gds.collections.ha.HugeIntArray labels,
 org.neo4j.gds.core.utils.paged.ReadOnlyHugeLongArray trainSet)

Specified by:

train in interface ClassifierTrainer