From 00182022e2471e8fa7873e89bd7e83e7fd4b7e2c Mon Sep 17 00:00:00 2001
From: mart-r <mart.ratas@gmail.com>
Date: Thu, 22 Feb 2024 17:30:49 +0000
Subject: [PATCH] CU-8693wtx4: Add option to filter by CUI when comparing

---
 medcat/compare_models/model_comparison.ipynb | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/medcat/compare_models/model_comparison.ipynb b/medcat/compare_models/model_comparison.ipynb
index c9eb022..2f5b4b4 100644
--- 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)"
    ]
   },
   {