hex.genmodel.attributes

Class VariableImportances

java.lang.Object

hex.genmodel.attributes.VariableImportances

All Implemented Interfaces:

java.io.Serializable

public class VariableImportances
extends java.lang.Object
implements java.io.Serializable

Represents model's variables and their relative importances in the model. The structure is model-independent.

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
double[]	_importances
java.lang.String[]	_variables

Constructor Summary

Constructors

Constructor and Description
VariableImportances(java.lang.String[] variableNames, double[] relativeImportances)

Method Summary

Modifier and Type	Method and Description
protected static VariableImportances	extractFromJson(com.google.gson.JsonObject modelJson)
int	numberOfUsedVariables()
KeyValue[]	topN(int n)

Methods inherited from class java.lang.Object

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

Field Detail

_variables

public final java.lang.String[] _variables

_importances

public final double[] _importances

Constructor Detail

VariableImportances

public VariableImportances(java.lang.String[] variableNames,
                           double[] relativeImportances)

Method Detail

extractFromJson

protected static VariableImportances extractFromJson(com.google.gson.JsonObject modelJson)

topN

public KeyValue[] topN(int n)

Parameters:

n - how many variables is in the output. If n >= number of variables or n <= 0 then all variables are returned.

Returns:

descending sorted array of String -> double. Where String is variable and double is relative importance of the variable.

numberOfUsedVariables

public int numberOfUsedVariables()