今回調べたこと ナイーブベイズを実装しようとした際に、最終的な予測の出力で迷った部分があったので、sklearnの実装を参考。 その際に実装の中身を追ったので、内容を記録。 ソースコード 参考コード 今回はBernoulliNBの実装を参考に中身を確認。 import numpy as np from sklearn.naive_bayes import BernoulliNB rng = np.random.RandomState(1) X = rng.randint(5, size=(6, 100)) Y = np.array([1, 2, 3, 4, 4, 5]) clf = BernoulliNB(force_alpha=True) clf.fit(X, Y) print(clf.predict(X[2:3]))