tidylearn (version 0.1.0)

tl_stratified_models: Stratified Features via Clustering

Description

Create cluster-specific supervised models for heterogeneous data

Usage

tl_stratified_models(
  data,
  formula,
  cluster_method = "kmeans",
  k = 3,
  supervised_method = "linear",
  ...
)

Value

A list of models (one per cluster) plus cluster assignments

Arguments

data

A data frame

formula

Model formula

cluster_method

Clustering method

k

Number of clusters

supervised_method

Supervised learning method

...

Additional arguments

Examples

# \donttest{
models <- tl_stratified_models(mtcars, mpg ~ ., cluster_method = "kmeans",
                                k = 3, supervised_method = "linear")
# }