hex.tree.xgboost.predict

Class XGBoostNativeVariableImportance

java.lang.Object

hex.tree.xgboost.predict.XGBoostNativeVariableImportance

All Implemented Interfaces:

XGBoostVariableImportance

public class XGBoostNativeVariableImportance
extends java.lang.Object
implements XGBoostVariableImportance

Constructor Summary

Constructors

Constructor and Description
XGBoostNativeVariableImportance(water.Key modelKey, java.lang.String featureMap)

Method Summary

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

Methods inherited from class java.lang.Object

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

Constructor Detail

XGBoostNativeVariableImportance

public XGBoostNativeVariableImportance(water.Key modelKey,
                                       java.lang.String featureMap)

Method Detail

cleanup

public void cleanup()

Specified by:

cleanup in interface XGBoostVariableImportance

getFeatureScores

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

Specified by:

getFeatureScores in interface XGBoostVariableImportance

parseFeatureScores

public static java.util.Map<java.lang.String,FeatureScore> parseFeatureScores(java.lang.String[] modelDump)