Skip to contents

Create an ensemble of three TorchDecon models with different architectures.

Usage

CreateTorchDeconEnsemble(n_features, n_classes, device = "auto", seed = NULL)

Arguments

n_features

Integer. Number of input features (genes).

n_classes

Integer. Number of output classes (cell types).

device

Character. Device to use. Default is "auto".

seed

Integer. Random seed. Default is NULL.

Value

A TorchDeconEnsemble object containing three models.

Examples

if (FALSE) { # \dontrun{
ensemble <- CreateTorchDeconEnsemble(n_features = 5000, n_classes = 10)
} # }