Ok, can’t see `prob`

getting `argmax()`

ed, but I think there’s more to this problem.

Can’t remember what the `F_score`

function accepted, but I guess it was one-hot encoded labels.

### Go on

Since you have numeric labels, my best guess would be to just one-hot encode them in this function (along with `argmax()`

ed `prob`

) and then apply the calculations.

### Revert

Also I suppose, you were closer to the solution back when you had binary cross entropy. But the missing one-hot encoded labels were a bit misleading

You probably have sigmoid as your last layer, which would work with one-hot encoded labels, BCE and this function.

Sorry if I introduced some confusion here, I’ve assumed that you do a slightly different approach