Skip to contents

Predict from a fitted cnn model

Source: R/cnn.R
predict.citocnn.Rd

Predict from a fitted cnn model

Usage

# S3 method for citocnn
predict(
  object,
  newdata = NULL,
  type = c("link", "response", "class"),
  device = c("cpu", "cuda", "mps"),
  ...
)

Arguments

object

a model created by cnn

newdata

new data for predictions

type

which value should be calculated, either raw response, output of link function or predicted class (in case of classification)

device

device on which network should be trained on.

...

additional arguments

Value

prediction matrix