Tabular classification training issues and probability output with fastai v2

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/

1 Like