参考课程中的应用实例,从生活中或人工智能开发案例中查找实例,用python设计一个应用“朴素贝叶斯分类器”或“决策树分类器”的模型。
时间: 2024-12-21 07:21:44 浏览: 19
朴素贝叶斯分类器是一种基于概率统计的简单而强大的机器学习算法,常用于文本分类、垃圾邮件识别等场景。这里我们可以设计一个简单的Python示例,使用scikit-learn库来创建一个文本分类器,比如预测一封电子邮件是否是垃圾邮件。
```python
# 导入必要的库
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.naive_bayes import MultinomialNB
from sklearn.model_selection import train_test_split
import pandas as pd
# 假设我们有一个包含两类(非垃圾邮件和垃圾邮件)的数据集,如email_data.csv
data = pd.read_csv('email_data.csv')
# 分割数据为特征和目标变量
X = data['text']
y = data['label']
# 文本预处理(例如分词)
vectorizer = CountVectorizer()
X_transformed = vectorizer.fit_transform(X)
# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X_transformed, y, test_size=0.2, random_state=42)
# 使用朴素贝叶斯分类器
nb_classifier = MultinomialNB()
nb_classifier.fit(X_train, y_train)
# 预测新邮件是否是垃圾邮件
new_email = "这是一封新的电子邮件..."
new_email_transformed = vectorizer.transform([new_email])
prediction = nb_classifier.predict(new_email_transformed)
print(f"新邮件属于{prediction[0]}类别.")
#
阅读全文