Skip to content

Latest commit

 

History

History
 
 

b_NaiveBayes

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 

Naive Bayes

Naive Bayes algorithms implemented with

  • np.bincount for MultinomialNB
  • np.exp for GaussianNB

MultinomialNB + GaussianNB = MergedNB

Visualization

Categorical Features

Numerical Features

Example

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