tl_stratified_models: Stratified Features via Clustering

In tidylearn: A Unified Tidy Interface to R's Machine Learning Ecosystem

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",
  ...
)

Arguments

data	A data frame
formula	Model formula
cluster_method	Clustering method
k	Number of clusters
supervised_method	Supervised learning method
...	Additional arguments

Value

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

Examples

models <- tl_stratified_models(mtcars, mpg ~ ., cluster_method = "kmeans",
                                k = 3, supervised_method = "linear")

tidylearn documentation built on Feb. 6, 2026, 5:07 p.m.