🔧 update value counts for predictions, fix bowang-lab#135

nato16 · Dec 12, 2023 · e881fd9 · e881fd9
1 parent ab566c7
commit e881fd9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tutorials/Tutorial_Reference_Mapping.ipynb b/tutorials/Tutorial_Reference_Mapping.ipynb
@@ -249,8 +249,8 @@
     "        idx = labels[k]\n",
     "    else:\n",
     "        idx, sim = get_similar_vectors(test_emebd[k][np.newaxis, ...], ref_cell_embeddings, k)\n",
-    "    pred = mode(ref_embed_adata.obs[cell_type_key][idx], axis=0)\n",
-    "    preds.append(pred[0][0])\n",
+    "    pred = ref_embed_adata.obs[cell_type_key][idx].value_counts()\n",
+    "    preds.append(pred.index[0])\n",
     "gt = test_adata.obs[cell_type_key].to_numpy()\n",
     "\n",
     "sklearn.metrics.accuracy_score(gt, preds)"