search

YubiaoYue / MedMamba

This is the official code repository for "MedMamba: Vision Mamba for Medical Image Classification"
297 stars 24 forks source link

Class Probability #22

Closed ericleonardo closed 2 months ago

ericleonardo commented 2 months ago

Hi! Please, will MedMamba output predicted class Probability? It is common for medical image classification to use probability to know if prediction is reliable.

Thank you and congratulations for the work!

YubiaoYue commented 2 months ago

Hi! MedMamba outputs predicted class Probability.