scmg.model.cell_type_search module

class scmg.model.cell_type_search.CellTypeSearcher(ref_cell_emb_adata, emb_key='X_scmg')

Bases: object

search_ref_cell(query_emb, project_umap=True, n_jobs=32)

Search for the reference cells that are most similar to each query cell. :param query_emb: query embedding of shape (n_cells, n_features) :type query_emb: np.array

search_ref_cell_types(query_emb, radius=2, n_jobs=32)

Search for reference cell types that are most similar to the query cells :param query_emb: query embedding of shape (n_cells, n_features) :type query_emb: np.array :param radius: soft search radius :type radius: float