CU-8693wtx4: Add option to filter by CUI when comparing

CogStack · Feb 22, 2024 · 0018202 · 0018202
1 parent 8efdf62
commit 0018202
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/medcat/compare_models/model_comparison.ipynb b/medcat/compare_models/model_comparison.ipynb
@@ -127,7 +127,11 @@
     "from compare import get_diffs_for\n",
     "from output import parse_and_show, show_dict_deep, compare_dicts\n",
     "\n",
-    "cdb_comp, tally1, tally2, ann_diffs = get_diffs_for(model_path_1, model_path_2, documents_file)"
+    "cui_filter = None\n",
+    "# optional CUI filter:\n",
+    "# cui_filter = {\"25064002\"}\n",
+    "\n",
+    "cdb_comp, tally1, tally2, ann_diffs = get_diffs_for(model_path_1, model_path_2, documents_file, cui_filter=cui_filter)"
    ]
   },
   {