single_partial_dependence_plot: Plot a single partial dependence plot

Description Usage Arguments Value

View source: R/random-forest-model.R

Description

Plot a single partial dependence plot

Usage

1
2
single_partial_dependence_plot(rf_model, data, variable, return_plot = FALSE,
  ylabel = "y")

Arguments

rf_model

random forest model obtained from the function fit_random_forest.

data

data used to fit the random forest model

variable

variable to use. A character.

return_plot

Boolean. If TRUE an object of class ggplot is returned, if FALSE the data about partial dependence is calculated and returned. FALSE by default.

ylabel

label of the y axis. Character. Defaults to "y".

Value

Data on partial dependence for the selected variable or a ggplot object depending on the argument return_plot.


pegoraro/qchlorophyll documentation built on May 24, 2019, 11:46 p.m.