Open
Description
- OS & Version: Windows 2016
- ML.NET Version: ML.NET 1.7.0
- .NET Version: .NET 5.0
Issue
Could not find column 'Probability' exception is thrown when transforming input data with model trained with NaiveBayes MultiClassClassification trainer.
Expected behavior
Similar to other trainers, Probability column representing probability for predicted label or array of probabilities representing probability for every label should be present in output of transformation.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment