наївний Баєсів класифікатор

import numpy as np

class NaiveBayesClassifier:

    def __init__(self):

        self.classes = None

        self.class_priors = None

        self.feature_probs = None

    def fit(self, X, y):

        self.classes, class_counts = np.unique(y, return_counts=True)

        self.class_priors = class_counts / y.size

        self.feature_probs = []

        for c in self.classes:

            X_c = X[y == c]

            feature_probs_c = np.mean(X_c, axis=0)

            self.feature_probs.append(feature_probs_c)

    def predict(self, X):

        posteriors = []

        for i, c in enumerate(self.classes):

            class_prior = np.log(self.class_priors[i])

            feature_probs_c = self.feature_probs[i]

            log_likelihoods = np.log(feature_probs_c)

            log_likelihoods_complement = np.log(1 - feature_probs_c)

            log_posteriors = np.sum(X * log_likelihoods + (1 - X) * log_likelihoods_complement, axis=1) + class_prior

            posteriors.append(log_posteriors)

        posteriors = np.array(posteriors).T

        predictions = np.argmax(posteriors, axis=1)

        return self.classes[predictions]

# Приклад використання

# Згенеруємо випадкові дані

X = np.random.randint(2, size=(1000, 10))  # Ознаки (0 або 1)

y = np.random.randint(2, size=1000)  # Мітки (0 або 1)

# Ініціалізуємо і навчаємо класифікатор

classifier = NaiveBayesClassifier()

classifier.fit(X, y)

# Передбачення на нових даних

new_data = np.random.randint(2, size=(5, 10))

predictions = classifier.predict(new_data)

print(predictions)