hex.tree.xgboost

Class XGBoostOutput

java.lang.Object

water.Iced

hex.Model.Output

hex.tree.xgboost.XGBoostOutput

All Implemented Interfaces:

hex.Model.GetNTrees, hex.tree.CalibrationHelper.OutputWithCalibration, java.io.Externalizable, java.io.Serializable, java.lang.Cloneable, water.Freezable

public class XGBoostOutput
extends hex.Model.Output
implements hex.Model.GetNTrees, hex.tree.CalibrationHelper.OutputWithCalibration

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
hex.Model<?,?,?>	_calib_model
water.util.TwoDimTable	_native_parameters
int	_ntrees
hex.ScoreKeeper[]	_scored_train
hex.ScoreKeeper[]	_scored_valid
boolean	_sparse
long[]	_training_time_ms
water.util.TwoDimTable	_variable_importances
water.util.TwoDimTable	_variable_importances_cover
water.util.TwoDimTable	_variable_importances_frequency
XgbVarImp	_varimp

Fields inherited from class hex.Model.Output

_column_types, _cross_validation_fold_assignment_frame_id, _cross_validation_holdout_predictions_frame_id, _cross_validation_metrics, _cross_validation_metrics_summary, _cross_validation_models, _cross_validation_predictions, _cv_scoring_history, _defaultThreshold, _distribution, _domains, _end_time, _hasFold, _hasOffset, _hasTreatment, _hasWeights, _isSupervised, _job, _model_summary, _modelClassDist, _names, _orig_projection_array, _origDomains, _origNames, _priorClassDist, _reproducibility_information_table, _run_time, _scoring_history, _start_time, _total_run_time, _training_metrics, _validation_metrics

Constructor Summary

Constructors

Constructor and Description
XGBoostOutput(XGBoost b)

Method Summary

Modifier and Type	Method and Description
hex.Model<?,?,?>	calibrationModel()
water.util.TwoDimTable	createInputFramesInformationTable(hex.ModelBuilder modelBuilder)
int	getInformationTableNumRows()
int	getNTrees()
water.util.TwoDimTable	getVariableImportances()
hex.ScoreKeeper[]	scoreKeepers()
void	setCalibrationModel(hex.Model<?,?,?> model)

Methods inherited from class hex.Model.Output

changeModelMetricsKey, checksum_impl, classNames, clearModelMetrics, defaultThreshold, features, foldIdx, foldName, getModelCategory, getModelMetrics, hasFold, hasOffset, hasResponse, hasTreatment, hasWeights, interactionBuilder, isAutoencoder, isBinomialClassifier, isClassifier, isMultinomialClassifier, isSupervised, lastSpecialColumnIdx, nclasses, nfeatures, offsetIdx, offsetName, printTwoDimTables, resetThreshold, responseIdx, responseName, setNames, setNames, startClock, stopClock, toString, treatmentIdx, weightsIdx, weightsName

Methods inherited from class water.Iced

asBytes, clone, copyOver, frozenType, read, readExternal, readJSON, reloadFromBytes, toJsonBytes, toJsonString, write, writeExternal, writeJSON

Methods inherited from class java.lang.Object

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

Methods inherited from interface hex.tree.CalibrationHelper.OutputWithCalibration

getCalibrationMethod, getModelCategory, isCalibrated

Field Detail

_sparse

public boolean _sparse

_ntrees

public int _ntrees

_scored_train

public hex.ScoreKeeper[] _scored_train

_scored_valid

public hex.ScoreKeeper[] _scored_valid

_training_time_ms

public long[] _training_time_ms

_variable_importances

public water.util.TwoDimTable _variable_importances

_variable_importances_cover

public water.util.TwoDimTable _variable_importances_cover

_variable_importances_frequency

public water.util.TwoDimTable _variable_importances_frequency

_varimp

public XgbVarImp _varimp

_native_parameters

public water.util.TwoDimTable _native_parameters

_calib_model

public hex.Model<?,?,?> _calib_model

Constructor Detail

XGBoostOutput

public XGBoostOutput(XGBoost b)

Method Detail

scoreKeepers

public hex.ScoreKeeper[] scoreKeepers()

getVariableImportances

public water.util.TwoDimTable getVariableImportances()

Overrides:

getVariableImportances in class hex.Model.Output

createInputFramesInformationTable

public water.util.TwoDimTable createInputFramesInformationTable(hex.ModelBuilder modelBuilder)

Overrides:

createInputFramesInformationTable in class hex.Model.Output

getInformationTableNumRows

public int getInformationTableNumRows()

Overrides:

getInformationTableNumRows in class hex.Model.Output

getNTrees

public int getNTrees()

Specified by:

getNTrees in interface hex.Model.GetNTrees

calibrationModel

public hex.Model<?,?,?> calibrationModel()

Specified by:

calibrationModel in interface hex.tree.CalibrationHelper.OutputWithCalibration

setCalibrationModel

public void setCalibrationModel(hex.Model<?,?,?> model)

Specified by:

setCalibrationModel in interface hex.tree.CalibrationHelper.OutputWithCalibration