I am trying to use Naive Bayes for a project of mine in Python and I want to obtain the probability value after having built the model.
Suppose I have two classes - A and B. Currently there is an API to to find which class a sample belongs to (predict). Now, I want to find the probability of it belonging to Class A or Class B. Can you please provide any details about how I can obtain the probability values for it belonging to the either class A or class B?