单类分类的python代码示例
时间: 2023-10-08 18:14:50 浏览: 82
好的,以下是单类分类的Python代码示例:
```python
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.naive_bayes import MultinomialNB
from sklearn.metrics import accuracy_score
# 读取数据集
data = pd.read_csv('data.csv')
# 分割训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(data['text'], data['label'], random_state=0)
# 对文本进行特征提取
vect = CountVectorizer().fit(X_train)
X_train_vectorized = vect.transform(X_train)
# 使用多项式朴素贝叶斯模型进行训练
model = MultinomialNB()
model.fit(X_train_vectorized, y_train)
# 测试模型准确率
X_test_vectorized = vect.transform(X_test)
y_pred = model.predict(X_test_vectorized)
accuracy = accuracy_score(y_test, y_pred)
print('Accuracy:', accuracy)
```
请问还有其他问题需要我回答吗?