Garfield.model.utils.weighted_knn_trainer

Garfield.model.utils.weighted_knn_trainer(train_adata, train_adata_emb, n_neighbors=50)[source]

Trains a weighted KNN classifier on train_adata.

Parameters:

train_adata (AnnData) – Annotated dataset to be used to train KNN classifier with label_key as the target variable.
train_adata_emb (str) – Name of the obsm layer to be used for calculation of neighbors. If set to “X”, anndata.X will be used
n_neighbors (int) – Number of nearest neighbors in KNN classifier.