plotFeatureImportance: Plot feature importance

Description Usage Arguments Value

View source: R/plotFeatureImportance.R

Description

Plot feature importance

Usage

1
2
3
4
5
6
7
8
plotFeatureImportance(
  df,
  final.model.imp = NULL,
  title = NULL,
  mid.function = "mean",
  sort = T,
  export.features = F
)

Arguments

final.model.imp

If a vector is provided, red dashes will be added to the plot according to vector values. Used for overlaying the feature importance of the final model over feature importance as determined by cross-validation

title

User specified plot title. If unspecified, title defaults to 'Feature importance'

sort

Whether to sort the features by decreasing importance

export.features

If TRUE, only returns a vector containing the feature names

df.importance

A data.frame containing a column of feature names and a column of MeanDecreaseAccuracy values

Value

Returns a ggplot2 scatter plot


luannnguyen/mltoolkit documentation built on Aug. 29, 2020, 8:31 a.m.