生成一个贝叶斯的分类器的代码
时间: 2024-05-05 18:15:52 浏览: 68
基于python实现训练高斯朴素贝叶斯分类器进行分类源码+项目说明+注释(可自动生成C测试代码).zip
以下是一个简单的贝叶斯分类器的Python代码示例:
```python
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.naive_bayes import GaussianNB
from sklearn.metrics import accuracy_score
# 读取数据集
data = pd.read_csv("data.csv")
# 划分数据集
X = data.drop("label", axis=1)
y = data["label"]
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 创建并训练模型
model = GaussianNB()
model.fit(X_train, y_train)
# 预测并评估模型
y_pred = model.predict(X_test)
accuracy = accuracy_score(y_test, y_pred)
print("Accuracy:", accuracy)
```
以上代码假设数据集已经存在于名为“data.csv”的文件中,并且标签列名为“label”。它使用GaussianNB类创建一个高斯朴素贝叶斯分类器,并在训练集上拟合模型。然后,该模型用于预测测试集的标签,并计算分类器的准确度。
阅读全文