predict.citocnn: Predict from a fitted cnn model

In cito: Building and Training Neural Networks

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

cito documentation built on May 29, 2024, 1:28 a.m.