Categories
Misc

Classification model outputs floats

Hi

I’m not sure what’s going on with my classification model, as it predicts objects as floats instead of classes. Is it something to do with the loss function or activation functions at the end of the network?

The barebones code can be found here:

https://pastebin.com/wd8i6P7R

submitted by /u/dahkneela
[visit reddit] [comments]

Leave a Reply

Your email address will not be published. Required fields are marked *