org.apache.ignite.ml.trainers

Interface DatasetTrainer<M extends Model,L>

Type Parameters:

M - Type of a produced model.

L - Type of a label.

All Known Subinterfaces:

MultiLabelDatasetTrainer<M>, SingleLabelDatasetTrainer<M>

All Known Implementing Classes:

DecisionTreeClassificationTrainer, DecisionTreeRegressionTrainer, KMeansTrainer, KNNClassificationTrainer, LinearRegressionLSQRTrainer, LinearRegressionSGDTrainer, MLPTrainer, SVMLinearBinaryClassificationTrainer, SVMLinearMultiClassClassificationTrainer

public interface DatasetTrainer<M extends Model,L>

Interface for trainers. Trainer is just a function which produces model from the data.

Method Summary

Modifier and Type	Method and Description
<K,V> M	fit(DatasetBuilder<K,V> datasetBuilder, IgniteBiFunction<K,V,double[]> featureExtractor, IgniteBiFunction<K,V,L> lbExtractor) Trains model based on the specified data.
default <K,V> M	fit(Ignite ignite, IgniteCache<K,V> cache, IgniteBiFunction<K,V,double[]> featureExtractor, IgniteBiFunction<K,V,L> lbExtractor) Trains model based on the specified data.
default <K,V> M	fit(Map<K,V> data, int parts, IgniteBiFunction<K,V,double[]> featureExtractor, IgniteBiFunction<K,V,L> lbExtractor) Trains model based on the specified data.

Method Detail

fit

<K,V> M fit(DatasetBuilder<K,V> datasetBuilder,
            IgniteBiFunction<K,V,double[]> featureExtractor,
            IgniteBiFunction<K,V,L> lbExtractor)

Trains model based on the specified data.

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.

fit

default <K,V> M fit(Ignite ignite,
                    IgniteCache<K,V> cache,
                    IgniteBiFunction<K,V,double[]> featureExtractor,
                    IgniteBiFunction<K,V,L> lbExtractor)

Trains model based on the specified data.

Type Parameters:

K - Type of a key in upstream data.

V - Type of a value in upstream data.

Parameters:

ignite - Ignite instance.

cache - Ignite cache.

featureExtractor - Feature extractor.

lbExtractor - Label extractor.

Returns:

Model.

fit

default <K,V> M fit(Map<K,V> data,
                    int parts,
                    IgniteBiFunction<K,V,double[]> featureExtractor,
                    IgniteBiFunction<K,V,L> lbExtractor)

Trains model based on the specified data.

Type Parameters:

K - Type of a key in upstream data.

V - Type of a value in upstream data.

Parameters:

data - Data.

parts - Number of partitions.

featureExtractor - Feature extractor.

lbExtractor - Label extractor.

Returns:

Model.