python中sklearn的朴素贝叶斯方法(sklearn.naive_bayes.GaussianNB)的简单使用

时间:2025-02-15 18:19:31
#测试数据
import numpy as np
features_train = ([[-1, -1], [-2, -1], [-3, -2], [1, 1], [2, 1], [3, 2]])
labels_train = ([1, 1, 1, 2, 2, 2])
#引入高斯朴素贝叶斯
from sklearn.naive_bayes import GaussianNB
#实例化
clf = GaussianNB()
#训练数据 fit相当于train
(features_train, labels_train) 
#输出单个预测结果
features_test = ([-0.8,-1])
labels_test = ([1])
pred = (features_test)
print(pred)
#准确度评估 评估正确/总数
#方法1
accuracy = (features_test, labels_test)
#方法2
from  import accuracy_score
accuracy2 = accuracy_score(pred,labels_test)

更多参考:/stable/modules/generated/sklearn.naive_bayes.