朴素贝叶斯
贝叶斯也是一个分类算法。在本篇文章中,和决策树一样,首先先介绍一下朴素
贝叶斯算法步骤,简单做一个例题,最后用Python简单实现一下。
1、朴素贝叶斯的介绍:
2、简简单单的例题:
3、Python实现一下:
代码如下:
from sklearn.datasets import load_iris #导入数据集
from sklearn.naive_bayes import GaussianNB #朴素贝叶斯分类器#
from sklearn.model_selection import train_test_split
iris = load_iris()
# 解析数据
x=iris.data
y=iris.target.reshape(-1,1)
# 划分训练集和测试集
x_train,x_test,y_train,y_test=train_test_split(x,y,test_size=0.3,random_state=30,stratify=y)
module = GaussianNB()
module.fit(x_train, y_train.ravel())
y_predi = module.predict(x_test)
ans = 0
for i in range(len(y_test)):
if y_test[i] == y_predi[i]:
ans += 1
print(ans / len(y_test))