PyTorch implementation of PBA.
Project description
torch-pba
PyTorch Implementation of PBA. AnnData-centric.
Installation
From PYPI:
pip install torch-pba
Alternatively, install the development version from GitHub:
git clone https://github.com/mvinyard/torch-pba.git; cd torch-pba; pip install -e .
Example use:
from torch_pba import PBA
from anndata import read_h5ad
pba = PBA(adata=read_h5ad("./path/to/adata.h5ad"))
pba.build_kNN()
pba.compute_Laplacian()
pba.compute_potential()
pba.compute_fate_bias()
pba.compute_mean_first_passage_time()
Time to calculate Mean First Passage Time for the example hematopoiesis dataset is cut from 4+ hours to <10 mins. In this example, I used a NVIDIA T4 GPU rented from GCP.
See more: notebook
Original work:
Note:
I have not contributed any methodological novelty in this library. The original implementation contains the novel application of a Laplace transform to a kNN Graph to obtain a potential value, pseudotime, etc. Here, I have simply adapted the library to PyTorch/CUDA. No formal benchmarking has been performed.
Contact / questions:
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
Built Distribution
Hashes for torch_pba-0.0.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 981f424cca4f3e96a38f9be314c968d53b270cdd14bf7eaecea3f8a578e29439 |
|
MD5 | a6f27f1112c9c0dd4948a7aabbf5094a |
|
BLAKE2b-256 | e3f042a4f1a8ba6976be0a974317d868587dda13d5054cc95b1c05b6aac1d9fe |