hex.genmodel.algos.tree

Class SharedTreeMojoModelWithContributions

java.lang.Object

hex.genmodel.GenModel

hex.genmodel.MojoModel

hex.genmodel.algos.tree.SharedTreeMojoModel

hex.genmodel.algos.tree.SharedTreeMojoModelWithContributions

All Implemented Interfaces:

SharedTreeGraphConverter, IGenModel, PredictContributionsFactory, java.io.Serializable, IGeneratedModel

Direct Known Subclasses:

DrfMojoModel, GbmMojoModel

public abstract class SharedTreeMojoModelWithContributions
extends SharedTreeMojoModel
implements PredictContributionsFactory

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
class	SharedTreeMojoModelWithContributions.ContributionsPredictor

Nested classes/interfaces inherited from class hex.genmodel.algos.tree.SharedTreeMojoModel

SharedTreeMojoModel.DecisionPathTracker<T>, SharedTreeMojoModel.LeafDecisionPathTracker, SharedTreeMojoModel.LeafNodeAssignments, SharedTreeMojoModel.StringDecisionPathTracker

Field Summary

Fields inherited from class hex.genmodel.algos.tree.SharedTreeMojoModel

_calib_glm_beta, _compressed_trees, _compressed_trees_aux, _ntree_groups, _ntrees_per_group, _variable_importances

Fields inherited from class hex.genmodel.MojoModel

_balanceClasses, _category, _defaultThreshold, _h2oVersion, _modelAttributes, _modelClassDistrib, _modelDescriptor, _mojo_version, _nclasses, _nfeatures, _priorClassDistrib, _supervised, _uuid

Fields inherited from class hex.genmodel.GenModel

_domains, _names, _offsetColumn, _responseColumn

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	SharedTreeMojoModelWithContributions(java.lang.String[] columns, java.lang.String[][] domains, java.lang.String responseColumn)

Method Summary

Modifier and Type	Method and Description
protected abstract SharedTreeMojoModelWithContributions.ContributionsPredictor	getContributionsPredictor(TreeSHAPPredictor<double[]> treeSHAPPredictor)
double	getInitF()
PredictContributions	makeContributionsPredictor() Create an instance of PredictContributions The returned implementation is not guaranteed to be thread-safe and the caller is responsible for making sure each thread will have own copy of the instance

Methods inherited from class hex.genmodel.algos.tree.SharedTreeMojoModel

_computeGraph, calibrateClassProbabilities, computeTreeGraph, convert, getDecisionPath, getDecisionPath, getDecisionPath, getDecisionPathNames, getLeafNodeAssignments, getLeafNodeId, getNTreeGroups, getNTreesPerGroup, postInit, scoreAllTrees, scoreSingleTree, scoreStagedPredictions, scoreTree, scoreTree, scoreTree0, scoreTree0, scoreTree1, scoreTree1, scoreTreeRange, treeName, unifyPreds

Methods inherited from class hex.genmodel.MojoModel

getModelCategory, getUUID, isSupervised, load, load, nclasses, nfeatures

Methods inherited from class hex.genmodel.GenModel

bitSetContains, bitSetIsInRange, convertDouble2Float, correctProbabilities, createAuxKey, features, GBM_rescale, getColIdx, getDomainValues, getDomainValues, getDomainValues, getHeader, getModelCategories, getNames, getNumClasses, getNumCols, getNumResponseClasses, getPrediction, getPredsSize, getPredsSize, getResponseIdx, getResponseName, GLM_identityInv, GLM_inverseInv, GLM_logInv, GLM_logitInv, GLM_ologitInv, GLM_tweedieInv, img2pixels, isAutoEncoder, isClassifier, KMeans_closest, KMeans_distance, KMeans_distance, KMeans_distances, Kmeans_preprocessData, Kmeans_preprocessData, KMeans_simplex, log_rescale, mapEnum, score0, score0, setCats, setCats, setInput, setInput

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

SharedTreeMojoModelWithContributions

protected SharedTreeMojoModelWithContributions(java.lang.String[] columns,
                                               java.lang.String[][] domains,
                                               java.lang.String responseColumn)

Method Detail

makeContributionsPredictor

public PredictContributions makeContributionsPredictor()

Description copied from interface: PredictContributionsFactory

Create an instance of PredictContributions The returned implementation is not guaranteed to be thread-safe and the caller is responsible for making sure each thread will have own copy of the instance

Specified by:

makeContributionsPredictor in interface PredictContributionsFactory

Returns:

instance of PredictContributions

getInitF

public double getInitF()

getContributionsPredictor

protected abstract SharedTreeMojoModelWithContributions.ContributionsPredictor getContributionsPredictor(TreeSHAPPredictor<double[]> treeSHAPPredictor)