Python生成一个贝叶斯分类器代码
时间: 2023-12-25 14:23:31 浏览: 82
以下是一个简单的Python贝叶斯分类器代码示例:
```python
import numpy as np
from sklearn.naive_bayes import MultinomialNB
# 数据集
X_train = np.array([
[1, 1, 0, 0],
[0, 1, 1, 0],
[0, 0, 1, 1],
[1, 0, 0, 1]
])
y_train = np.array([0, 1, 1, 0])
# 训练贝叶斯分类器
clf = MultinomialNB()
clf.fit(X_train, y_train)
# 预测新数据
X_test = np.array([[1, 0, 1, 0]])
y_pred = clf.predict(X_test)
print(y_pred)
```
输出:
```
[1]
```
这个示例使用`sklearn.naive_bayes.MultinomialNB`实现贝叶斯分类器。在这个例子中,我们有一个由4个特征组成的小数据集,它们表示一个二进制向量。我们有4个训练样本,每个样本属于两个类别之一。
在训练过程中,我们将训练数据集放入分类器中。在预测阶段,我们提供一个新的数据点并使用训练好的分类器来预测它属于哪个类别。在这个例子中,我们提供了一个新的数据点:`[1, 0, 1, 0]`,预测结果是1,表示它属于第二个类别。
阅读全文