org.apache.ignite.ml.regressions.linear

Class LinearRegressionModel

java.lang.Object

org.apache.ignite.ml.regressions.linear.LinearRegressionModel

All Implemented Interfaces:

Serializable, Function<Vector,Double>, Exportable<LinearRegressionModel>, IgniteFunction<Vector,Double>, Model<Vector,Double>

public class LinearRegressionModel
extends Object
implements Model<Vector,Double>, Exportable<LinearRegressionModel>, Serializable

Simple linear regression model which predicts result value Y as a linear combination of input variables: Y = weights * X + intercept.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
LinearRegressionModel(Vector weights, double intercept)

Method Summary

Modifier and Type	Method and Description
Double	apply(Vector input)
boolean	equals(Object o)
double	getIntercept()
Vector	getWeights()
int	hashCode()
<P> void	saveModel(Exporter<LinearRegressionModel,P> exporter, P path) Save model by the given path.
String	toString()
String	toString(boolean pretty)

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface org.apache.ignite.ml.Model

combine

Methods inherited from interface java.util.function.Function

andThen, compose, identity

Constructor Detail

LinearRegressionModel

public LinearRegressionModel(Vector weights,
                             double intercept)

Method Detail

getWeights

public Vector getWeights()

getIntercept

public double getIntercept()

apply

public Double apply(Vector input)

Specified by:

apply in interface Function<Vector,Double>

saveModel

public <P> void saveModel(Exporter<LinearRegressionModel,P> exporter,
                          P path)

Save model by the given path.

Specified by:

saveModel in interface Exportable<LinearRegressionModel>

Parameters:

exporter - Exporter.

path - Path to saved model.

equals

public boolean equals(Object o)

Overrides:

equals in class Object

hashCode

public int hashCode()

Overrides:

hashCode in class Object

toString

public String toString()

Overrides:

toString in class Object

toString

public String toString(boolean pretty)

Specified by:

toString in interface Model<Vector,Double>

Parameters:

pretty - Use pretty mode.