高斯貝葉斯分類器


import numpy as np
from sklearn.naive_bayes import GaussianNB #導入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為高斯朴素貝葉斯分類器
clf.fit(X, Y) #訓練數據

print(clf.predict([[-1, -1]])) #預測數據[-1,0]屬於哪一類


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM