jsat.classifiers.trees

Class ImportanceByUses

java.lang.Object

jsat.classifiers.trees.ImportanceByUses

All Implemented Interfaces:

java.io.Serializable, TreeFeatureImportanceInference

public class ImportanceByUses
extends java.lang.Object
implements TreeFeatureImportanceInference

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
ImportanceByUses()
ImportanceByUses(boolean weightByDepth)

Method Summary

Modifier and Type	Method and Description
<Type extends DataSet> double[]	getImportanceStats(TreeLearner model, DataSet<Type> data)

Methods inherited from class java.lang.Object

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

Constructor Detail

ImportanceByUses

public ImportanceByUses(boolean weightByDepth)

ImportanceByUses

public ImportanceByUses()

Method Detail

getImportanceStats

public <Type extends DataSet> double[] getImportanceStats(TreeLearner model,
                                                          DataSet<Type> data)

Specified by:

getImportanceStats in interface TreeFeatureImportanceInference

Parameters:

model - the tree model to infer feature importance from

data - the dataset to use for importance inference. Should be either a Classification or Regression dataset, depending on the type of the model.

Returns:

a double array with one entry for each feature. Numeric features start first, followed by categorical features. Larger values indicate higher importance, and all values must be non-negative. Otherwise, no constraints are placed on the output of this function.