Create Dataset for Training

Create a torch dataset from expression matrix.

Usage

MakeDataset(X, loss_mode = "nb")

Arguments

X

Expression matrix (cells x genes)

loss_mode

Loss mode: "mse", "nb", or "zinb"

Value

torch::dataset object