Skip to content
This repository was archived by the owner on Jul 28, 2025. It is now read-only.

Commit f0b8d94

Browse files
committed
CU-8697v6qr2 add logging suggested by the review
1 parent b7f5efe commit f0b8d94

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

medcat/ner/transformers_ner.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -344,8 +344,10 @@ def expand_model_with_concepts(self, cui2preferred_name: Dict[str, str], use_avg
344344
avg_weight = torch.mean(self.model.classifier.weight, dim=0, keepdim=True)
345345
avg_bias = torch.mean(self.model.classifier.bias, dim=0, keepdim=True)
346346

347+
new_cuis = set()
347348
for label, preferred_name in cui2preferred_name.items():
348349
if label in self.model.config.label2id.keys():
350+
logger.warning("Concept ID '%s' already exists in the model, skipping...", label)
349351
continue
350352

351353
sname = preferred_name.lower().replace(" ", "~")
@@ -382,6 +384,10 @@ def expand_model_with_concepts(self, cui2preferred_name: Dict[str, str], use_avg
382384
self.model.num_labels += 1
383385
self.model.classifier.out_features += 1
384386

387+
new_cuis.add(label)
388+
389+
logger.info("Model expanded with the new concept(s): %s and shall be retrained before use.", str(new_cuis))
390+
385391
@classmethod
386392
def load(cls, save_dir_path: str, config_dict: Optional[Dict] = None) -> "TransformersNER":
387393
"""Load a meta_cat object.

0 commit comments

Comments
 (0)