get_disp_feats {treeheatr}

Select the important features to be displayed.

Description

Select features with p-value (computed from decision tree) < 'p_thres' or all features if 'show_all_feats == TRUE'.

Usage

get_disp_feats(fit, feat_names, show_all_feats, p_thres)

Arguments

fit

constparty object of the decision tree.

feat_names

Character vector specifying the feature names in dat.

show_all_feats

Logical. If TRUE, show all features regardless of 'p_thres'.

p_thres

Numeric value indicating the p-value threshold of feature importance. Feature with p-values computed from the decision tree below this value will be displayed on the heatmap.

Value

A character vector of feature names.