chanzuckerberg / cellxgene-census

CZ CELLxGENE Discover Census
https://chanzuckerberg.github.io/cellxgene-census/
MIT License
88 stars 22 forks source link

PyTorch support for stratified subsampling #479

Open atolopko-czi opened 1 year ago

Ivana-Jelic commented 1 year ago

Hi Andrew, is this to going to address: https://github.com/chanzuckerberg/cellxgene-census/issues/304

atolopko-czi commented 1 year ago

Hi Andrew, is this to going to address: https://github.com/chanzuckerberg/cellxgene-census/issues/304

This issue is to capture that stratified subsampling support should be available via PyTorch DataLoaders (actually DataPipes). But I expect the stratified subsampling implementation to be a general feature of the Census API, which the PyTorch code delegates to. So I would keep this issue open and separate for now.