omicverse.utils.weighted_knn_trainer¶

omicverse.utils.weighted_knn_trainer(train_adata, train_adata_emb, n_neighbors=50)[source]¶

Trains a weighted KNN classifier on train_adata.

Arguments: train_adata: :Annotated dataset to be used to train KNN classifier with label_key as the target variable. train_adata_emb: Name of the obsm layer to be used for calculation of neighbors. If set to “X”, anndata.X will be used n_neighbors: Number of nearest neighbors in KNN classifier.
Returns: k_neighbors_transformer: KNeighborsTransformer