predict.citommn: Predict from a fitted mmn model

In citoverse/cito: Building and Training Neural Networks

Predict from a fitted mmn model

Description

Predict from a fitted mmn model

Usage

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

Arguments

object	a model created by mmn
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

citoverse/cito documentation built on Jan. 16, 2025, 11:49 p.m.