鸢尾花python贝叶斯分类_高斯朴素贝叶斯方法进行鸢尾花分类

贝叶斯方法完整代码

import seaborn as sns

iris = sns.load_dataset('iris')

X_iris = iris.drop('species', axis=1)

y_iris = iris['species']

print(X_iris)

from sklearn.cross_validation import train_test_split

Xtrain, Xtest, ytrain, ytest = train_test_split(X_iris, y_iris, random_state=1)

from sklearn.naive_bayes import GaussianNB

model = GaussianNB()

model.fit(Xtrain, ytrain)

y_model = model.predict(Xtest)

from sklearn.metrics import accuracy_score

accuracy_score(ytest, y_model)

步骤分析

一-首先获取数据.

这里我们在线导入seaborn库的iris(鸢尾花)数据

import seaborn as sns

iris = sns.load_dataset('iris')

这是github上的说明, 可直接下载csv文档

二-将数据格式化

X_iris = iris.drop('species', axis=1)