naive_bayes_construct.py

import numpy as np


class NaiveBayes:

    def __init__(self, X, y):
        self.priors = []  # List of priors - P(y) for each class
        self.stds = []  # List of STDs of feature columns
        self.means = []  # List of Mean of feature columns
        self.classes = np.unique(y)  # unique class labels

        self.X = X
        self.y = y