org.apache.ignite.ml.composition.boosting

Class GDBTrainer.GDBModel

java.lang.Object

org.apache.ignite.ml.composition.ModelsComposition

org.apache.ignite.ml.composition.boosting.GDBTrainer.GDBModel

All Implemented Interfaces:

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

Enclosing class:

GDBTrainer

public static class GDBTrainer.GDBModel
extends ModelsComposition

GDB model.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
GDBModel(List<? extends Model<Vector,Double>> models, WeightedPredictionsAggregator predictionsAggregator, IgniteFunction<Double,Double> internalToExternalLblMapping) Creates an instance of GDBModel.

Method Summary

Modifier and Type	Method and Description
Double	apply(Vector features) Applies containing models to features and aggregate them to one prediction.

Methods inherited from class org.apache.ignite.ml.composition.ModelsComposition

getModels, getPredictionsAggregator, saveModel, toString, toString

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

GDBModel

public GDBModel(List<? extends Model<Vector,Double>> models,
                WeightedPredictionsAggregator predictionsAggregator,
                IgniteFunction<Double,Double> internalToExternalLblMapping)

Creates an instance of GDBModel.

Parameters:

models - Models.

predictionsAggregator - Predictions aggregator.

internalToExternalLblMapping - Internal to external lbl mapping.

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>

Overrides:

apply in class ModelsComposition

Parameters:

features - Features vector.

Returns:

Estimation.