scvi-tools
scvi-tools copied to clipboard
Some suggestion about GNN
Is your feature request related to a problem? Please describe. I want to develop a new model with Pytorch Geometric which is a GNN library based on Pytorch. I hope you can develop some api to make it more easier.
Describe the solution you'd like
- Please develop a obsm_field for AnnDataManager to register obsm(n_obs*n_obs) attribute. Because scanpy neighbors save the graph adjacency matrix in obsm as key "connectivities".
- DataSplitter in scvi.dataloaders maybe also should be updated to support minibatch operation. This minibatch operation is similar to subgraph extraction.
Thank you for the request.
We are interested in this but have no estimated timeline for when this would be available.