Package org.neo4j.gds.ml.models.linearregression

Class LinearRegressionObjective

java.lang.Object

org.neo4j.gds.ml.models.linearregression.LinearRegressionObjective

All Implemented Interfaces:

Objective<LinearRegressionData>

public class LinearRegressionObjective
extends Object
implements Objective<LinearRegressionData>

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)
LinearRegressionData	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

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<LinearRegressionData>

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<LinearRegressionData>

modelData

public LinearRegressionData 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<LinearRegressionData>

Returns:

the data