GaussianNB

Предполагает, что признаки внутри каждого класса распределены нормально и независимы. Очень быстрый, хорош как базовая модель, особенно при малом числе данных.

from sklearn.naive_bayes import GaussianNB

clf = GaussianNB()
clf.fit(X_train, y_train)
print(clf.predict(X_test))