PDF says softmax 4 but create_COVIDx_v2.ipynb only creates 3 categories.

lindawangg / COVID-Net

COVID-Net Open Source Initiative

Other

1.15k stars 480 forks source link

PDF says softmax 4 but create_COVIDx_v2.ipynb only creates 3 categories. #31

Closed Justus-M closed 4 years ago

Justus-M commented 4 years ago

Apparently the model uses 4 softmax layers, but the latest dataset creation notebook only splits the cases into Normal, Pneumonia and Covid-19. There seems to be a mismatch here. What is the latest approach?

lindawangg commented 4 years ago

This is an error. Softmax should be 3 classes. Thanks for pointing it out.