iamDecode / sklearn-pmml-model

A library to parse and convert PMML models into Scikit-learn estimators.
BSD 2-Clause "Simplified" License
76 stars 15 forks source link

Correctly parse priors in naive bayes estimators #28

Closed iamDecode closed 3 years ago

iamDecode commented 3 years ago

The prior probabilities in Naive Bayes estimators was ignored and assumed to be uniform (1/n classes). This fix ensures the correct priors are parsed, and will assume a uniform distribution in absence of a BayesOutput element.