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

Class MLPClassifier

java.lang.Object

org.neo4j.gds.ml.models.mlp.MLPClassifier

All Implemented Interfaces:

Classifier

public final class MLPClassifier
extends Object
implements Classifier

Nested Class Summary

Nested classes/interfaces inherited from interface org.neo4j.gds.ml.models.Classifier

Classifier.ClassifierData

Constructor Summary

Constructors

Constructor	Description
MLPClassifier(MLPClassifierData data)

Method Summary

Modifier and Type	Method	Description
MLPClassifierData	data()
double[]	predictProbabilities(double[] features)
org.neo4j.gds.ml.core.tensor.Matrix	predictProbabilities(org.neo4j.gds.ml.core.batch.Batch batch, Features features)

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.neo4j.gds.ml.models.Classifier

numberOfClasses

Constructor Details

MLPClassifier

public MLPClassifier(MLPClassifierData data)

Method Details

predictProbabilities

public double[] predictProbabilities(double[] features)

Specified by:

predictProbabilities in interface Classifier

predictProbabilities

public org.neo4j.gds.ml.core.tensor.Matrix predictProbabilities(org.neo4j.gds.ml.core.batch.Batch batch,
 Features features)

Specified by:

predictProbabilities in interface Classifier

data

public MLPClassifierData data()

Specified by:

data in interface Classifier