Hi @ptrblck , thank you. This works.
However, do you know how I can attain the prediction values back from the regression model given that the targets are actually categories in my case? Would merely rounding the output of the of the regression model work?
For instance, my targets are [1,2,4,2,2,1,3,1] as my model output is [0.23, -0.12, 0.13, 0.17, 0.71, -0.26, 0.45, 0.37]. Do you know how I can attain the predictions back in the form of the targets?