details_logistic_reg_keras: Logistic regression via keras

details_logistic_reg_kerasR Documentation

Logistic regression via keras


keras_mlp() fits a generalized linear model for binary outcomes. A linear combination of the predictors is used to model the log odds of an event.


For this engine, there is a single mode: classification

Tuning Parameters

This model has one tuning parameter:

  • penalty: Amount of Regularization (type: double, default: 0.0)

For penalty, the amount of regularization is only L2 penalty (i.e., ridge or weight decay).

Translation from parsnip to the original package

logistic_reg(penalty = double(1)) %>% 
  set_engine("keras") %>% 
## Logistic Regression Model Specification (classification)
## Main Arguments:
##   penalty = double(1)
## Computational engine: keras 
## Model fit template:
## parsnip::keras_mlp(x = missing_arg(), y = missing_arg(), penalty = double(1), 
##     hidden_units = 1, act = "linear")

keras_mlp() is a parsnip wrapper around keras code for neural networks. This model fits a linear regression as a network with a single hidden unit.

Preprocessing requirements

Factor/categorical predictors need to be converted to numeric values (e.g., dummy or indicator variables) for this engine. When using the formula method via fit(), parsnip will convert factor columns to indicators.

Predictors should have the same scale. One way to achieve this is to center and scale each so that each predictor has mean zero and a variance of one.

Case weights

The underlying model implementation does not allow for case weights.

Saving fitted model objects

Models fitted with this engine may require native serialization methods to be properly saved and/or passed between R sessions. To learn more about preparing fitted models for serialization, see the bundle package.


The “Fitting and Predicting with parsnip” article contains examples for logistic_reg() with the "keras" engine.


  • Hoerl, A., & Kennard, R. (2000). Ridge Regression: Biased Estimation for Nonorthogonal Problems. Technometrics, 42(1), 80-86.

parsnip documentation built on June 24, 2024, 5:14 p.m.