hex.genmodel

Interface IGenModel

All Known Implementing Classes:

CoxPHMojoModel, DeeplearningMojoModel, DrfMojoModel, ExtendedIsolationForestMojoModel, GamMojoModel, GamMojoModelBase, GamMojoMultinomialModel, GbmMojoModel, GenModel, GlmMojoModel, GlmMojoModelBase, GlmMultinomialMojoModel, GlmOrdinalMojoModel, GlrmMojoModel, IsolationForestMojoModel, IsotonicRegressionMojoModel, KLimeMojoModel, KMeansMojoModel, MojoModel, MojoPipeline, PCAMojoModel, RuleFitMojoModel, SharedTreeMojoModel, SharedTreeMojoModelWithContributions, StackedEnsembleMojoModel, SvmMojoModel, TargetEncoderMojoModel, UpliftDrfMojoModel, Word2VecMojoModel

public interface IGenModel

Interface publishing methods for generated models. This interface extend the original interface from H2O.

Method Summary

Modifier and Type	Method and Description
java.lang.String[]	features() Returns names of input features.
java.util.EnumSet<ModelCategory>	getModelCategories() For models with multiple categories, returns the set of all supported categories.
ModelCategory	getModelCategory() Returns this model category.
boolean	isSupervised() Returns true for supervised models.
int	nclasses() Returns number of output classes for classifiers or 1 for regression models.
int	nfeatures() Returns number of input features.

Method Detail

isSupervised

boolean isSupervised()

Returns true for supervised models.

Returns:

true if this class represents supervised model.

nfeatures

int nfeatures()

Returns number of input features.

Returns:

number of input features used for training.

features

java.lang.String[] features()

Returns names of input features.

Returns:

names of input features used for training.

nclasses

int nclasses()

Returns number of output classes for classifiers or 1 for regression models. For unsupervised models returns 0.

Returns:

returns number of output classes or 1 for regression models.

getModelCategory

ModelCategory getModelCategory()

Returns this model category.

Returns:

model category

See Also:

ModelCategory

getModelCategories

java.util.EnumSet<ModelCategory> getModelCategories()

For models with multiple categories, returns the set of all supported categories.