Naive Bayes algorithm is a simple yet powerful machine learning algorithm. It is used in python machine learning that can be used for both binary and multiclass classification problems.

What is Naive Bayes Algorithm?

The Naive Bayes algorithm is a probabilistic algorithm that makes use of Bayes’ theorem to predict the class of a target variable based on the values of predictor variables. It is called “naive” because it makes an assumption of independence between the features of the data. Despite this assumption, it performs well in practice for many datasets.

Types of Naive Bayes Algorithm

There are three main types of Naive Bayes algorithms: Gaussian, Multinomial, and Bernoulli. Each of these algorithms makes use of a different probability distribution to model the relationship between the target and predictor variables.

Gaussian Naive Bayes

The Gaussian Naive Bayes algorithm assumes that the continuous predictor variables are normally distributed. This makes it ideal for datasets where the predictor variables are continuous.

Multinomial Naive Bayes

The Multinomial Naive Bayes algorithm is suitable for datasets where the predictor variables are discrete, such as text data. It models the relationship between the target and predictor variables using a multinomial distribution.

Bernoulli Naive Bayes

The Bernoulli Naive Bayes algorithm is similar to the Multinomial Naive Bayes algorithm, but is specifically designed for datasets where the predictor variables are binary.

Implementing Naive Bayes in Python

The Python library scikit-learn provides an implementation of the Naive Bayes algorithm. Let’s look at an example of how to use the Gaussian Naive Bayes algorithm to classify the iris dataset.

import pandas as pd
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.naive_bayes import GaussianNB
from sklearn.metrics import accuracy_score

# Load the iris dataset
iris = load_iris()
X =
y =

# Split the data into training and testing sets
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)

# Train the Gaussian Naive Bayes model
gnb = GaussianNB(), y_train)

# Predict the target for the test set
y_pred = gnb.predict(X_test)

# Calculate the accuracy of the model
accuracy = accuracy_score(y_test, y_pred)
print("Accuracy: ", accuracy)

In this example, we load the iris dataset and split it into training and testing sets. We then train a Gaussian Naive Bayes model on the training data and use it to predict the target for the test set. Finally, we calculate the accuracy of the model.


The Naive Bayes algorithm is a simple and effective method for classification and prediction in supervised learning. It is easy to implement in Python using the scikit-learn library. Whether you are working with continuous or discrete data, there is a Naive Bayes algorithm to suit your needs.

Also check WHAT IS GIT ? It’s Easy If You Do It Smart

You can also visite the Git website (

One Response

Leave a Reply

Your email address will not be published. Required fields are marked *