I changed it to a target column with labels, used CrossEntropyLossFlat
as the loss function, and removed y_range
. That did the trick. Thank you!
For anyone who comes across this in the future, this is a useful post: https://machinelearningmastery.com/how-to-choose-loss-functions-when-training-deep-learning-neural-networks/