Skip to contents

Softmax regression

Source: R/multinom-reg.R
softmax_regression.Rd

A wrapper around mlpack::softmax_regression() that allows passing a formula.

Usage

softmax_regression(
  formula = NULL,
  data = NULL,
  penalty = 0.001,
  epochs = 400,
  no_intercept = FALSE,
  x = NULL,
  y = NULL
)

Arguments

formula

A formula.

data

A data.frame.

penalty

L2-regularization constant.

epochs

Maximum number of iterations.

no_intercept

Logical; passed to mlpack::softmax_regression().

x

Design matrix.

y

Response matrix.

Value

An object of class baritsu_sr.

See also

mlpack::softmax_regression() predict.baritsu_sr()