fit_linear_model_tree: Fit Linear Model tree to 'klassets_xy' object

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

Fit Linear Model tree to klassets_xy object

Description

Fit Linear Model tree to klassets_xy object

Usage

fit_linear_model_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::mob_control.
alpha	Alpha value, same used in partykit::mob_control
...	Additional options passed to partykit::mob_control.

Examples

df <- sim_xy()

df

dflm <- fit_linear_model_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_linear_model_tree(df))

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