# -*- coding: utf-8 -*-
# @Date    : 2019-05-20 17:00:43
# @Author  : QilongPan 
# @Email   : 3102377627@qq.com
import numpy as np 
from sklearn.naive_bayes import GaussianNB

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

clf = GaussianNB()
clf.fit(X,Y)
print(clf.predict([[-0.8,-1]]))