Style transfer variational autoencoder
Project description
The official pytorch implementation of “Style transfer with variational autoencoders is a promising approach to RNA-Seq data harmonization and analysis”. The package contains a code for training and testing the model, as well as a code for working with different types of datasets.
Installation
To install the latest version from PyPI, use:
>>> pip install stvae
Benchmarks
The original code containing code with testing several models can be found here.
Example
ds = stvae.datasets.MouseDataset()
cfg = stvae.Config()
train, test, classif = ds.split(0.15, True, 0.15)
cfg.count_classes = ds.n_labels
cfg.count_classes = ds.n_batches
cfg.input_dim = ds.nb_genes
model = stvae.stVAE(cfg)
model.train(train, None)
d = model.test(test, classif)
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
stVAE-0.1b0.tar.gz
(31.0 kB
view hashes)