hex.tree.xgboost.predict

Class XGBoostJavaVariableImportance

java.lang.Object

hex.tree.xgboost.predict.XGBoostJavaVariableImportance

All Implemented Interfaces:

XGBoostVariableImportance

public class XGBoostJavaVariableImportance
extends java.lang.Object
implements XGBoostVariableImportance

Constructor Summary

Constructors

Constructor and Description
XGBoostJavaVariableImportance(XGBoostModelInfo modelInfo)

Method Summary

Modifier and Type	Method and Description
java.util.Map<java.lang.String,FeatureScore>	getFeatureScores(byte[] boosterBytes)

Methods inherited from class java.lang.Object

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

Methods inherited from interface hex.tree.xgboost.predict.XGBoostVariableImportance

cleanup

Constructor Detail

XGBoostJavaVariableImportance

public XGBoostJavaVariableImportance(XGBoostModelInfo modelInfo)

Method Detail

getFeatureScores

public java.util.Map<java.lang.String,FeatureScore> getFeatureScores(byte[] boosterBytes)

Specified by:

getFeatureScores in interface XGBoostVariableImportance