init model

Init Cite-seq model with given hyperparams.

Parameters

model: CiteAutoencoder CiteAutoencoder model object.

---

device: str Device to run the training on (CPU or Cuda if GPU is available).

---

learning_rate: float Adjustable learning rate for improving gradient descent optimizer.

---

epochs: int Number of epochs to do the training.

---

optimizer: str Optimizer for neural network.

---

train_split: int Percntage of dataset to be used for training (not testing/validation).

---

latent_dimensions: int Dimensionality of latent space in autoencoder.

---

hidden_layers: int Number of hidden layers in autoencoder model.

---

Web view