Classification model outputs floats


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:

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

Leave a Reply

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