BiomedSciAI / fuse-med-ml

A python framework accelerating ML based discovery in the medical field by encouraging code reuse. Batteries included :)
Apache License 2.0
137 stars 34 forks source link

Matching keys lengths and new metrics #358

Closed IdoAmosIBM closed 5 months ago

IdoAmosIBM commented 5 months ago

Added 2 new metrics: mathews correlation coef. and balanced accuracy. Added a new functionality to default collate allowing to match the length of some keys in the batch dict to other keys - example: for encoder model, matching the length of the labels to the length of the input after crop padding is applied.