CogStack / MedCAT

Medical Concept Annotation Tool
Other
454 stars 105 forks source link

MetaCAT fixes and upgrades #495

Closed shubham-s-agarwal closed 1 month ago

shubham-s-agarwal commented 1 month ago

Pushing for 3 updates:

  1. Removed the check and update for labels with zero data, as this was causing issues during evaluation
  2. Resolved an issue where the confusion matrix couldn't be calculated when testing on a single class with an F1 score of 1, as it expected the original number of training classes (3)
  3. Updated the attention mask creation to dynamically use the actual pad_idx value instead of assuming it to be 0