org.apache.ignite.ml.regressions.linear

Class LinearRegressionLSQRTrainer

java.lang.Object

org.apache.ignite.ml.regressions.linear.LinearRegressionLSQRTrainer

All Implemented Interfaces:

DatasetTrainer<LinearRegressionModel,Double>, SingleLabelDatasetTrainer<LinearRegressionModel>

public class LinearRegressionLSQRTrainer
extends Object
implements SingleLabelDatasetTrainer<LinearRegressionModel>

Trainer of the linear regression model based on LSQR algorithm.

See Also:

AbstractLSQR

Constructor Summary

Constructors

Constructor and Description
LinearRegressionLSQRTrainer()

Method Summary

Modifier and Type	Method and Description
<K,V> LinearRegressionModel	fit(DatasetBuilder<K,V> datasetBuilder, IgniteBiFunction<K,V,double[]> featureExtractor, IgniteBiFunction<K,V,Double> lbExtractor) Trains model based on the specified data.

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.ignite.ml.trainers.DatasetTrainer

fit, fit

Constructor Detail

LinearRegressionLSQRTrainer

public LinearRegressionLSQRTrainer()

Method Detail

fit

public <K,V> LinearRegressionModel fit(DatasetBuilder<K,V> datasetBuilder,
                                       IgniteBiFunction<K,V,double[]> featureExtractor,
                                       IgniteBiFunction<K,V,Double> lbExtractor)

Trains model based on the specified data.

Specified by:

fit in interface DatasetTrainer<LinearRegressionModel,Double>

Type Parameters:

K - Type of a key in upstream data.

V - Type of a value in upstream data.

Parameters:

datasetBuilder - Dataset builder.

featureExtractor - Feature extractor.

lbExtractor - Label extractor.

Returns:

Model.