predict.stoch_logistic function

Prediction function for stochastic logistic regression

Prediction function for stochastic logistic regression

Makes predictions for new data from the fitted model. Model have already been fit to at least 1 batch of data.

## S3 method for class 'stoch_logistic' predict(object, newdata, type = "prob", ...)

Arguments

  • object: A stoch_logistic object as output by function stochastic.logistic.regression.
  • newdata: New data on which to make predictions.
  • type: Type of prediction to make. Can pass prob to get probabilities for the positive class, or class to get the predicted class.
  • ...: Not used.

Returns

A vector with the predicted classes or probabilities for newdata.

See Also

stochastic.logistic.regression

  • Maintainer: David Cortes
  • License: BSD_2_clause + file LICENSE
  • Last published: 2021-09-26