torch-adata
Project description
Create pytorch Datasets from AnnData
Installation
Install from PYPI:
pip install torch-adata
Install the developer version:
git clone https://github.com/mvinyard/torch-adata.git; cd torch-adata;
pip install -e .
Example use of the base class
The base class, AnnDataset
is a subclass of the widely-used torch.utils.data.Dataset
.
import anndata as a
import torch_adata
adata = a.read_h5ad("/path/to/data.h5ad")
dataset = torch_adata.AnnDataset(adata)
Returns sampled data X_batch
as a torch.Tensor
.
# create a dummy index
idx = np.random.choice(range(dataset.__len__()), 5)
X_batch = dataset[idx]
TimeResolvedAnnDataset
Specialized class for time-resolved datasets. A subclass of the class, AnnDataset
.
import anndata as a
import torch_adata as ta
adata = a.read_h5ad("/path/to/data.h5ad")
dataset = torch_adata.TimeResolvedAnnDataset(
adata,
time_key="Time point",
use_key="X_pca",
obs_key="weight_key",
return_t=True,
)
When return_t == True
:
X, y, t = dataset[idx]
When return_t == False
:
X, y = dataset[idx]
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
torch-adata-0.0.13.tar.gz
(6.2 kB
view hashes)
Built Distribution
Close
Hashes for torch_adata-0.0.13-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 0ba60cc632cc24fabcdfc38243fb61db576ad8c15095e821332a8407401fe00e |
|
MD5 | cfab230051f1e7e21a37e563f755878b |
|
BLAKE2b-256 | c6554ac76e2d2c8ccc59710241cc9f34e7ee1ab61236f965ea8f9f39dd814d7f |