fit_regression_tree: Fit regression tree to 'klassets_xy' object

In jbkunst/klassets: Tools to simulate data set to teach Statistical Models and ML Algorithms

Fit regression tree to klassets_xy object

Description

Fit regression tree to klassets_xy object

Usage

fit_regression_tree(df, maxdepth = Inf, alpha = 0.05, ...)

Arguments

df	A object from sim_response_xy.
maxdepth	Max depth of the tree. Same used in partykit::ctree_control.
alpha	Alpha value, same used in partykit::ctree_control
...	Options for partykit::ctree_control.

Examples

df <- sim_xy()

df

dflm <- fit_regression_tree(df)

dflm

plot(dflm)

df <- sim_xy(n = 1000, x_dist = runif)
df <- dplyr::mutate(df, y = y + 2*sin(5 * x))
plot(df)

plot(fit_regression_tree(df, maxdepth = 3))

# default
plot(fit_regression_tree(df))

jbkunst/klassets documentation built on Dec. 7, 2022, 9:18 p.m.