Skip to contents

Extract a model workflow predictions from a tidyAML model tibble.

Usage

extract_wflw_pred(.data, .model_id = NULL)

Arguments

.data

The model table that must have the class tidyaml_mod_spec_tbl.

.model_id

The model number that you want to select, Must be an integer or sequence of integers, ie. 1 or c(1,3,5) or 1:2

Value

A tibble with the chosen model workflow(s).

Details

This function allows you to get a model workflow predictions or more from a tibble with a class of "tidyaml_mod_spec_tbl". It allows you to select the model by the .model_id column. You can call the model id's by an integer or a sequence of integers.

See also

Author

Steven P. Sanderson II, MPH

Examples

library(recipes)

rec_obj <- recipe(mpg ~ ., data = mtcars)
frt_tbl <- fast_regression(mtcars, rec_obj, .parsnip_eng = c("lm","glm"),
                                           .parsnip_fns = "linear_reg")

extract_wflw_pred(frt_tbl, 1)
#> [[1]]
#> # A tibble: 24 × 1
#>    .pred
#>    <dbl>
#>  1  23.8
#>  2  15.5
#>  3  28.1
#>  4  27.0
#>  5  13.6
#>  6  22.0
#>  7  29.3
#>  8  28.5
#>  9  24.5
#> 10  10.3
#> # … with 14 more rows
#> 
extract_wflw_pred(frt_tbl, 1:2)
#> [[1]]
#> # A tibble: 24 × 1
#>    .pred
#>    <dbl>
#>  1  23.8
#>  2  15.5
#>  3  28.1
#>  4  27.0
#>  5  13.6
#>  6  22.0
#>  7  29.3
#>  8  28.5
#>  9  24.5
#> 10  10.3
#> # … with 14 more rows
#> 
#> [[2]]
#> # A tibble: 24 × 1
#>    .pred
#>    <dbl>
#>  1  23.8
#>  2  15.5
#>  3  28.1
#>  4  27.0
#>  5  13.6
#>  6  22.0
#>  7  29.3
#>  8  28.5
#>  9  24.5
#> 10  10.3
#> # … with 14 more rows
#>