Make sparse tensors csr instead of coo when `return_sparse_X=True` - Githubissues

chanzuckerberg / cellxgene-census

CZ CELLxGENE Discover Census

https://chanzuckerberg.github.io/cellxgene-census/

MIT License

72 stars 19 forks source link

Make sparse tensors csr instead of coo when `return_sparse_X=True` #1120

Open pablo-gar opened 2 months ago

pablo-gar commented 2 months ago

When this parameter is set to True, make the tensor csr instead of coo

https://github.com/chanzuckerberg/cellxgene-census/blob/34d4dc0dc08602f1b444b36c44ee50d59111f7ed/api/python/cellxgene_census/src/cellxgene_census/experimental/ml/pytorch.py#L229C9-L229C24

cc @martinkim0