Based on your code, I assume you are working on a multi-class classification use case.
However, the target seems to be a regression target?
Could you explain your use case and how the target is defined?