get_disp_feats: Select the important features to be displayed.

Description Usage Arguments Value

View source: R/draw_heat.R

Description

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

Usage

1
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.


treeheatr documentation built on Nov. 20, 2020, 1:07 a.m.