search

cellarium-ai / cellarium-ml

Distributed single-cell data analysis.
BSD 3-Clause "New" or "Revised" License
11 stars 2 forks source link

scvi - implement `predict()` #147

Closed sjfleming closed 5 months ago

sjfleming commented 5 months ago

Implement the predict() method of the scvi model. This is just creating the embeddings given raw input data.

sjfleming commented 5 months ago

Very likely it can just be

        return self.inference(
            x=x_ng,
            batch_index=batch_index_n,
            cont_covs=cont_covs_nc,
            cat_covs=cat_covs_nd,
            n_samples=1,
        )