dotnet / dotnet/machinelearning
'Probability' column missing in output schema of NaiveBayes MulticlassClassification trainer
Open
area-Trainers
bug
- Dominant language
- C#
- Stars
- 9.4k
- Forks
- 2k
- Avg merge
- 2d 20h
- Merged PRs (30d)
- 11
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.
Contributor guide
Assessment
This issue has not been assessed yet.