ArtsAutosBooksBusinessEducationEntertainmentFamilyFashionFoodGamesGenderHealthHolidaysHomeHubPagesPersonal FinancePetsPoliticsReligionSportsTechnologyTravel

Naïve Bayes in Python

Updated on July 12, 2012

The Naive Bayes Algorithm

The naïve Bayes algorithm is a classifier based on Bayes' theorem. It relies on independence between features, which sometimes necessitates pre-processing (for example, via eigenvalue decomposition). Formally, the algorithm operates under supervised learning.

As implemented below, input data (consisting of feature vectors and corresponding classification) is supplied to the constructor. Maximum likelihood estimation is used to compute the conditional probabilities under assumption of Normality (other probability densities are possible). In the example, examples are either 'A' or 'B' and the feature space consists of two dimensional floating points, however the model is not limited to binary classification and the feature space may be arbitrarily large. For convenience, a prediction method is included, as well as a string representation.

Python Implementation of Gaussian Naïve Bayes

from __future__ import division
from numpy import *
from scipy.stats import norm

class GaussianNaiveBayes:

    def __init__(self, training_set):
        self.training_set = training_set

        self.labels = {}
        for label in [t[1] for t in self.training_set]:
            if self.labels.has_key(label):
                self.labels[label] += 1
            else: self.labels[label] = 1

        self.probl = {}
        self.N = len(self.training_set[0][0])
        self.cmean = {}
        self.cstdv = {}
        self.prob = {}
        for label in self.labels.keys():
            self.probl[label] = self.labels[label]/len(self.training_set)
            example_set = [e[0] for e in self.training_set if e[1] == label]
            self.cmean[label] = [0 for i in range(self.N)]
            self.cstdv[label] = [0 for i in range(self.N)]
            self.prob[label] = [0 for i in range(self.N)]
            for i in range(self.N):
                xi = array([e[i] for e in example_set])
                self.cmean[label][i] = xi.mean()
                self.cstdv[label][i] = xi.std()
                self.prob[label][i] = lambda x: norm.pdf(x, xi.mean(), xi.std())

    def predict(self, X):
        p = {}
        for l in self.labels.keys():
            cp = prod([self.prob[l][i](X[i]) for i in range(self.N)])
            p[l] = self.probl[l]*cp
        results = [(p[l],l) for l in self.labels.keys()]
        return results[-1][1], results[-1][0]/sum([r[0] for r in results])

    def __str__(self):
        s = 'Gaussian Naive Bayes Model:\n'
        for l in self.labels.keys():
            s+= ' P(%s) = %.3f\n'%(l, self.probl[l])
            for i in range(self.N):
                s+= ' E(X%s|%s) = %.3f\n'%(i+1, l, self.cmean[l][i])
                s+= ' VAR(X%s|%s) = %.3f\n'%(i+1, l, self.cstdv[l][i]**2)
        return s.strip()

if __name__ == '__main__':

    examples = []

    nb = GaussianNaiveBayes(examples)
    print nb
    print nb.predict((1,3))

Applications of Naïve Bayes

Bayes' classifiers are popular in spam filtering and have also been used predict the incidence of diseases based on clinical features.


    0 of 8192 characters used
    Post Comment

    No comments yet.


    This website uses cookies

    As a user in the EEA, your approval is needed on a few things. To provide a better website experience, uses cookies (and other similar technologies) and may collect, process, and share personal data. Please choose which areas of our service you consent to our doing so.

    For more information on managing or withdrawing consents and how we handle data, visit our Privacy Policy at:

    Show Details
    HubPages Device IDThis is used to identify particular browsers or devices when the access the service, and is used for security reasons.
    LoginThis is necessary to sign in to the HubPages Service.
    Google RecaptchaThis is used to prevent bots and spam. (Privacy Policy)
    AkismetThis is used to detect comment spam. (Privacy Policy)
    HubPages Google AnalyticsThis is used to provide data on traffic to our website, all personally identifyable data is anonymized. (Privacy Policy)
    HubPages Traffic PixelThis is used to collect data on traffic to articles and other pages on our site. Unless you are signed in to a HubPages account, all personally identifiable information is anonymized.
    Amazon Web ServicesThis is a cloud services platform that we used to host our service. (Privacy Policy)
    CloudflareThis is a cloud CDN service that we use to efficiently deliver files required for our service to operate such as javascript, cascading style sheets, images, and videos. (Privacy Policy)
    Google Hosted LibrariesJavascript software libraries such as jQuery are loaded at endpoints on the or domains, for performance and efficiency reasons. (Privacy Policy)
    Google Custom SearchThis is feature allows you to search the site. (Privacy Policy)
    Google MapsSome articles have Google Maps embedded in them. (Privacy Policy)
    Google ChartsThis is used to display charts and graphs on articles and the author center. (Privacy Policy)
    Google AdSense Host APIThis service allows you to sign up for or associate a Google AdSense account with HubPages, so that you can earn money from ads on your articles. No data is shared unless you engage with this feature. (Privacy Policy)
    Google YouTubeSome articles have YouTube videos embedded in them. (Privacy Policy)
    VimeoSome articles have Vimeo videos embedded in them. (Privacy Policy)
    PaypalThis is used for a registered author who enrolls in the HubPages Earnings program and requests to be paid via PayPal. No data is shared with Paypal unless you engage with this feature. (Privacy Policy)
    Facebook LoginYou can use this to streamline signing up for, or signing in to your Hubpages account. No data is shared with Facebook unless you engage with this feature. (Privacy Policy)
    MavenThis supports the Maven widget and search functionality. (Privacy Policy)
    Google AdSenseThis is an ad network. (Privacy Policy)
    Google DoubleClickGoogle provides ad serving technology and runs an ad network. (Privacy Policy)
    Index ExchangeThis is an ad network. (Privacy Policy)
    SovrnThis is an ad network. (Privacy Policy)
    Facebook AdsThis is an ad network. (Privacy Policy)
    Amazon Unified Ad MarketplaceThis is an ad network. (Privacy Policy)
    AppNexusThis is an ad network. (Privacy Policy)
    OpenxThis is an ad network. (Privacy Policy)
    Rubicon ProjectThis is an ad network. (Privacy Policy)
    TripleLiftThis is an ad network. (Privacy Policy)
    Say MediaWe partner with Say Media to deliver ad campaigns on our sites. (Privacy Policy)
    Remarketing PixelsWe may use remarketing pixels from advertising networks such as Google AdWords, Bing Ads, and Facebook in order to advertise the HubPages Service to people that have visited our sites.
    Conversion Tracking PixelsWe may use conversion tracking pixels from advertising networks such as Google AdWords, Bing Ads, and Facebook in order to identify when an advertisement has successfully resulted in the desired action, such as signing up for the HubPages Service or publishing an article on the HubPages Service.
    Author Google AnalyticsThis is used to provide traffic data and reports to the authors of articles on the HubPages Service. (Privacy Policy)
    ComscoreComScore is a media measurement and analytics company providing marketing data and analytics to enterprises, media and advertising agencies, and publishers. Non-consent will result in ComScore only processing obfuscated personal data. (Privacy Policy)
    Amazon Tracking PixelSome articles display amazon products as part of the Amazon Affiliate program, this pixel provides traffic statistics for those products (Privacy Policy)