org.apache.ignite.ml.composition

Class ModelsComposition

java.lang.Object

org.apache.ignite.ml.composition.ModelsComposition

All Implemented Interfaces:

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

Direct Known Subclasses:

GDBTrainer.GDBModel

public class ModelsComposition
extends Object
implements Model<Vector,Double>, Exportable<ModelsCompositionFormat>

Model consisting of several models and prediction aggregation strategy.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
ModelsComposition(List<? extends Model<Vector,Double>> models, PredictionsAggregator predictionsAggregator) Constructs a new instance of composition of models.

Method Summary

Modifier and Type	Method and Description
Double	apply(Vector features) Applies containing models to features and aggregate them to one prediction.
List<Model<Vector,Double>>	getModels() Returns containing models.
PredictionsAggregator	getPredictionsAggregator() Returns predictions aggregator.
<P> void	saveModel(Exporter<ModelsCompositionFormat,P> exporter, P path) Save model by the given path.
String	toString()
String	toString(boolean pretty)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, 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

ModelsComposition

public ModelsComposition(List<? extends Model<Vector,Double>> models,
                         PredictionsAggregator predictionsAggregator)

Constructs a new instance of composition of models.

Parameters:

models - Basic models.

predictionsAggregator - Predictions aggregator.

Method Detail

apply

public Double apply(Vector features)

Applies containing models to features and aggregate them to one prediction.

Specified by:

apply in interface Function<Vector,Double>

Parameters:

features - Features vector.

Returns:

Estimation.

getPredictionsAggregator

public PredictionsAggregator getPredictionsAggregator()

Returns predictions aggregator.

getModels

public List<Model<Vector,Double>> getModels()

Returns containing models.

saveModel

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

Save model by the given path.

Specified by:

saveModel in interface Exportable<ModelsCompositionFormat>

Parameters:

exporter - Exporter.

path - Path to saved model.

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.