fda models

Function	Works
`tidypredict_fit()`, `tidypredict_sql()`, `parse_model()`	✔
`tidypredict_to_column()`	✗
`tidypredict_test()`	✗
`tidypredict_interval()`, `tidypredict_sql_interval()`	✗
`parsnip`	✔

mda::fda() fits flexible discriminant analysis models. Predicting with such a model runs the predictors through a regression fit, projects the result onto the discriminant variates, and turns the distance to each class centroid into a posterior probability. When the regression fit is linear in the predictors, the whole path collapses into one linear predictor per outcome class and the posterior probabilities are the softmax of those linear predictors.

That restricts support to the two linear regression methods: mda::polyreg() (the default) with degree = 1, and mda::gen.ridge(), which is what discrim_linear() uses. Higher polynomial degrees and the mda::mars() and mda::bruto() methods are not supported, and neither are the mixture models from mda::mda().

Because these models predict one probability per outcome class, tidypredict_fit() returns a named list of expressions, one for each class, rather than a single expression. Since the output is a list, tidypredict_to_column() and tidypredict_test() are not supported.

`tidypredict_` functions

model <- mda::fda(Species ~ ., data = iris)

Create the R formulas, one per class

fit <- tidypredict_fit(model)
names(fit)
#> [1] "setosa"     "versicolor" "virginica"
fit[["setosa"]]
#> 1/(1 + exp(-2.0408182734589 + (Sepal.Length * -1.56244815123166) + 
#>     (Sepal.Width * -4.46535048744003) + (Petal.Length * 4.79149521011471) + 
#>     (Petal.Width * 3.12508713229891) - (-15.7712902867565 + (Sepal.Length * 
#>     6.44363108028089) + (Sepal.Width * 12.3870583475537) + (Petal.Length * 
#>     -17.2922700522402) + (Petal.Width * -21.193933257467))) + 
#>     exp(-34.1997086671399 + (Sepal.Length * -4.88118292904923) + 
#>         (Sepal.Width * -7.9217078601137) + (Petal.Length * 12.5007748421255) + 
#>         (Petal.Width * 18.0688461251681) - (-15.7712902867565 + 
#>         (Sepal.Length * 6.44363108028089) + (Sepal.Width * 12.3870583475537) + 
#>         (Petal.Length * -17.2922700522402) + (Petal.Width * -21.193933257467))))

Add the predictions to the original table

library(dplyr)

iris %>%
  mutate(!!!tidypredict_fit(model)) %>%
  glimpse()
#> Rows: 150
#> Columns: 8
#> $ Sepal.Length <dbl> 5.1, 4.9, 4.7, 4.6, 5.0, 5.4, 4.6, 5.0, 4.4, 4.9, 5.4, 4.…
#> $ Sepal.Width  <dbl> 3.5, 3.0, 3.2, 3.1, 3.6, 3.9, 3.4, 3.4, 2.9, 3.1, 3.7, 3.…
#> $ Petal.Length <dbl> 1.4, 1.4, 1.3, 1.5, 1.4, 1.7, 1.4, 1.5, 1.4, 1.5, 1.5, 1.…
#> $ Petal.Width  <dbl> 0.2, 0.2, 0.2, 0.2, 0.2, 0.4, 0.3, 0.2, 0.2, 0.1, 0.2, 0.…
#> $ Species      <fct> setosa, setosa, setosa, setosa, setosa, setosa, setosa, s…
#> $ setosa       <dbl> 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, …
#> $ versicolor   <dbl> 1.424733e-22, 3.225424e-18, 6.041136e-20, 6.010084e-17, 5…
#> $ virginica    <dbl> 3.699975e-43, 9.159088e-38, 7.720058e-40, 7.067088e-36, 1…

Confirm that the results match the model’s predict() results

probs <- sapply(fit, \(f) rlang::eval_tidy(f, iris))
all.equal(unname(probs), unname(predict(model, iris, type = "posterior")))
#> [1] TRUE

parsnip

parsnip fitted models are also supported by tidypredict:

library(parsnip)
library(discrim)

p_model <- discrim_linear(penalty = 1) %>%
  set_engine("mda") %>%
  fit(Species ~ ., data = iris)

tidypredict_fit(p_model)[["virginica"]]
#> 1/(exp(-11.9575128198861 + (Sepal.Length * 5.80316342827773) + 
#>     (Sepal.Width * 10.802899579644) + (Petal.Length * -16.5217691675749) + 
#>     (Petal.Width * -17.3381617005549) - (-33.2183837927994 + 
#>     (Sepal.Length * -4.41029984072505) + (Sepal.Width * -6.75173810807833) + 
#>     (Petal.Length * 12.0386389001348) + (Petal.Width * 14.7915580954519))) + 
#>     exp(-2.27975663240432 + (Sepal.Length * -1.39286358755267) + 
#>         (Sepal.Width * -4.05116147156562) + (Petal.Length * 4.48313026744007) + 
#>         (Petal.Width * 2.546603605103) - (-33.2183837927994 + 
#>         (Sepal.Length * -4.41029984072505) + (Sepal.Width * -6.75173810807833) + 
#>         (Petal.Length * 12.0386389001348) + (Petal.Width * 14.7915580954519))) + 
#>     1)

Parse model spec

Here is an example of the model spec:

pm <- parse_model(model)
str(pm, 2)
#> List of 3
#>  $ general    :List of 4
#>   ..$ model  : chr "fda"
#>   ..$ version: num 2
#>   ..$ type   : chr "multiclass_regression"
#>   ..$ family : chr "multinomial"
#>  $ classes    : chr [1:3] "setosa" "versicolor" "virginica"
#>  $ class_terms:List of 3
#>   ..$ :List of 5
#>   ..$ :List of 5
#>   ..$ :List of 5
#>  - attr(*, "class")= chr [1:3] "parsed_model" "pm_multiclass_regression" "list"

fda models

tidypredict_ functions

parsnip

Parse model spec

`tidypredict_` functions