I have a multiclass classification problem for which Ludwig produces good models. However, it always predicts 1 or close to 1 in exactly one category and zero in everything else:

count 135257.000000
mean 0.999987
std 0.001505
min 0.754915
25% 1.000000
50% 1.000000
75% 1.000000
max 1.000000

In the above example, only two out 135,257 have a prediction value of less than 1.

I have trained dozens of these models, most with very successful results, but this is an issue in every case. I have tried changing the selection of input features and adjusting the confidence_penalty and robust_lambda parameters. Robust_lambda had a significant effect on the model performance, but still did not effect mean probability.

Has anyone else encountered this behavior, and does anyone have any other ideas how to fix it?

Thanks,
Craig