Package org.neo4j.gds.ml.models.mlp
Class MLPClassifier
java.lang.Object
org.neo4j.gds.ml.models.mlp.MLPClassifier
- All Implemented Interfaces:
Classifier
-
Nested Class Summary
Nested classes/interfaces inherited from interface org.neo4j.gds.ml.models.Classifier
Classifier.ClassifierData -
Constructor Summary
Constructors -
Method Summary
Modifier and TypeMethodDescriptiondata()double[]predictProbabilities(double[] features) org.neo4j.gds.ml.core.tensor.MatrixpredictProbabilities(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, waitMethods inherited from interface org.neo4j.gds.ml.models.Classifier
numberOfClasses
-
Constructor Details
-
MLPClassifier
-
-
Method Details
-
predictProbabilities
public double[] predictProbabilities(double[] features) - Specified by:
predictProbabilitiesin interfaceClassifier
-
predictProbabilities
public org.neo4j.gds.ml.core.tensor.Matrix predictProbabilities(org.neo4j.gds.ml.core.batch.Batch batch, Features features) - Specified by:
predictProbabilitiesin interfaceClassifier
-
data
- Specified by:
datain interfaceClassifier
-