predict.citocnn {cito}

Predict from a fitted cnn model

Description

Predict from a fitted cnn model

Usage

## S3 method for class '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