Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Are the ECGResnet34 outputs probabilities? #40

Open
mah533 opened this issue May 8, 2022 · 0 comments
Open

Are the ECGResnet34 outputs probabilities? #40

mah533 opened this issue May 8, 2022 · 0 comments

Comments

@mah533
Copy link

mah533 commented May 8, 2022

Hello,

If my understanding is right:
The dimension of the output of the model is (batch size * number of classes). Logically, I would expect the elements in the second dimension (with dimension of number of classes) sum up to 1, i.e., they should represent the probability of belonging to that class. So the index of the maximum of the elements would identify the class of the input.

but, in my case at least, the elements do not sum up to 1 and I believe one normalization step (softmax function) should be added
as the last layer to make the sum equal to 1.

Thanks for your input,

E

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant