Creating a modelpack

Author: antsh3k

medcat/1_create_model/create_modelpack/create_modelpack.py

@@ -0,0 +1,35 @@
+from medcat.vocab import Vocab
+from medcat.cdb import CDB
+from medcat.cat import CAT
+
+
+cdb_model_path = "data/medcat_models/cdb"
+cdb_model = cdb_model_path + "<NAME OF MODEL HERE>.dat"  # Change to specific cdb of interest
+
+vocab_model_path = "data/medcat_models/vocab"
+modelpack_path = "data/medcat_models/modelpack"
+modelpack_name = "<NAME OF output MODELPACK HERE>.dat"  # Change to the name of your model
+
+# Load cdb
+cdb = CDB.load(cdb_model)
+
+# Set cdb configuration
+# technically we already created this during the cdb creation
+cdb.config.ner['min_name_len'] = 2
+cdb.config.ner['upper_case_limit_len'] = 3
+cdb.config.general['spell_check'] = True
+cdb.config.linking['train_count_threshold'] = 10
+cdb.config.linking['similarity_threshold'] = 0.3
+cdb.config.linking['train'] = True
+cdb.config.linking['disamb_length_limit'] = 4
+cdb.config.general['full_unlink'] = True
+
+# Load vocab
+vocab = Vocab.load(vocab_model_path)
+
+# Initialise the model
+cat = CAT(cdb=cdb, config=cdb.config, vocab=vocab)
+
+# Create and save model pack
+cat.create_model_pack(save_dir_path=modelpack_path, model_pack_name=modelpack_name)
+