Repository navigation
'Probability' column missing in output schema of NaiveBayes MulticlassClassification trainer #6033
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.
Activity
maheshlokhande commentedon Jan 11, 2022
Can anyone suggest how scores can be converted into probabilities for NaiveBayes since we need probabilities for our use case?