Naive Bayes algorithms implemented with
np.bincount
forMultinomialNB
np.exp
forGaussianNB
MultinomialNB
+ GaussianNB
= MergedNB
- Dataset comes from UCI: Bank Marketing dataset
from Util.Util import DataUtil
from b_NaiveBayes.Vectorized.GaussianNB import GaussianNB
x, y = DataUtil.gen_xor(one_hot=False) # Get xor dataset. Notice that y should not be one-hot
nb = GaussianNB()
nb.fit(x, y) # Train GaussianNB
nb.evaluate(x, y) # Print out accuracy
nb.visualize2d(x, y) # Visualize result (2d)
nb.visualize() # Visualize distribution
nb.show_timing_log() # Show timing log