In [1]:

    

import numpy as np
from sklearn.naive_bayes import GaussianNB


    

In [7]:

    

X = np.array([[-1, -1], [-2, -1], [-3, -2], [1, 1], [2, 1], [3, 2]])
Y = np.array([1, 1, 1, 2, 2, 2])


    

In [8]:

    

clf = GaussianNB()


    

In [9]:

    

clf.fit(X, Y)


    

    
Out[9]:





GaussianNB()

In [10]:

    

print(clf.predict([[-0.8, -1]]))


    

    




[1]

In [11]:

    

print(clf.predict([[-2, -4]]))


    

    




[1]