org.apache.ignite.ml.regressions.linear

Class LinearRegressionSGDTrainer<P extends Serializable>

java.lang.Object

org.apache.ignite.ml.regressions.linear.LinearRegressionSGDTrainer<P>

All Implemented Interfaces:

DatasetTrainer<LinearRegressionModel,Double>, SingleLabelDatasetTrainer<LinearRegressionModel>

public class LinearRegressionSGDTrainer<P extends Serializable>
extends Object
implements SingleLabelDatasetTrainer<LinearRegressionModel>

Trainer of the linear regression model based on stochastic gradient descent algorithm.

Constructor Summary

Constructors

Constructor and Description
LinearRegressionSGDTrainer(UpdatesStrategy<? super MultilayerPerceptron,P> updatesStgy, int maxIterations, int batchSize, int locIterations, long seed) Constructs a new instance of linear regression SGD trainer.

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

LinearRegressionSGDTrainer

public LinearRegressionSGDTrainer(UpdatesStrategy<? super MultilayerPerceptron,P> updatesStgy,
                                  int maxIterations,
                                  int batchSize,
                                  int locIterations,
                                  long seed)

Constructs a new instance of linear regression SGD trainer.

Parameters:

updatesStgy - Update strategy.

maxIterations - Max number of iteration.

batchSize - Batch size.

locIterations - Number of local iterations.

seed - Seed for random generator.

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.