jsat.classifiers.trees

Class ImportanceByUses

java.lang.Object

jsat.classifiers.trees.ImportanceByUses

All Implemented Interfaces:

Serializable, TreeFeatureImportanceInference

public class ImportanceByUses
extends Object
implements TreeFeatureImportanceInference

Author:

Edward Raff

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

clone, equals, finalize, 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.