From 693220a4bbaec5dae87f881063f8a5bb7cc6b33c Mon Sep 17 00:00:00 2001 From: Can Ergen Date: Thu, 25 Jul 2024 09:21:25 -0700 Subject: [PATCH] Update _onclass.py to work in fast mode. --- popv/algorithms/_onclass.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/popv/algorithms/_onclass.py b/popv/algorithms/_onclass.py index bc56e89..259cc43 100644 --- a/popv/algorithms/_onclass.py +++ b/popv/algorithms/_onclass.py @@ -179,7 +179,7 @@ def predict(self, adata): pred_label = [train_model.i2co[ind] for ind in onclass_seen] pred_label_str = [clid_2_name[ind] for ind in pred_label] adata.obs[self.result_key] = pred_label_str - adata.obs["onclass_seen"] = pred_label_str + adata.obs[self.seen_result_key] = pred_label_str else: onclass_pred = train_model.Predict(corr_test_feature, use_normalize=False, refine=True, unseen_ratio=-1.0) pred_label = [train_model.i2co[ind] for ind in onclass_pred[2]]