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

Class MLPClassifierObjective

java.lang.Object

org.neo4j.gds.ml.models.mlp.MLPClassifierObjective

All Implemented Interfaces:

Objective<MLPClassifierData>

public class MLPClassifierObjective
extends Object
implements Objective<MLPClassifierData>

Constructor Summary

Constructors

Constructor	Description
MLPClassifierObjective(MLPClassifier classifier, Features features, org.neo4j.gds.collections.ha.HugeIntArray labels, double penalty, double focusWeight, double[] classWeights)

Method Summary

Modifier and Type	Method	Description
org.neo4j.gds.ml.core.Variable<org.neo4j.gds.ml.core.tensor.Scalar>	loss(org.neo4j.gds.ml.core.batch.Batch batch, long trainSize)
MLPClassifierData	modelData()	Returns the data, such as weights, needed to store or load the model
List<org.neo4j.gds.ml.core.functions.Weights<? extends org.neo4j.gds.ml.core.tensor.Tensor<?>>>	weights()

Methods inherited from class java.lang.Object

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

Constructor Details

MLPClassifierObjective

public MLPClassifierObjective(MLPClassifier classifier,
 Features features,
 org.neo4j.gds.collections.ha.HugeIntArray labels,
 double penalty,
 double focusWeight,
 double[] classWeights)

Method Details

weights

public List<org.neo4j.gds.ml.core.functions.Weights<? extends org.neo4j.gds.ml.core.tensor.Tensor<?>>> weights()

Specified by:

weights in interface Objective<MLPClassifierData>

loss

public org.neo4j.gds.ml.core.Variable<org.neo4j.gds.ml.core.tensor.Scalar> loss(org.neo4j.gds.ml.core.batch.Batch batch,
 long trainSize)

Specified by:

loss in interface Objective<MLPClassifierData>

modelData

public MLPClassifierData modelData()

Description copied from interface: Objective

Returns the data, such as weights, needed to store or load the model

Specified by:

modelData in interface Objective<MLPClassifierData>

Returns:

the data