train: Main Training Model

In ff98li/scRGNet: Perform scRNA-seq Analysis using Single-cell Graphical Neural Network (scGNN) Framework

Description

Train in main

Usage

train(
  epoch,
  train_loader,
  model,
  optimiser,
  regu_mat = NULL,
  hyperParams,
  device,
  EMflag = FALSE
)

Arguments

epoch	epoch
train_loader	A torch dataloader loaded with an scDataset object
model	A feature auto-encoder object
optimiser	Optimiser
regu_mat	LTMG regulation matrix
hyperParams	Hyper-parameters user defined to tune the model
device	hardware to train the model
EMflag	Emflag #' @references \insertRefscGNNscRGNet

ff98li/scRGNet documentation built on Jan. 14, 2022, 4:58 a.m.