如何使用Python的sklearn CountVectorizer?
作者:小确幸
时间:2023-04-24
来源:正软商城
浏览:5
简介CountVectorizer官方文档。将一个文档集合向量化为为一个计数矩阵。如果不提供一个先验字典,不使用分析器做某种特征选择,那么特征的数量将等于通过分析数据发现的词汇量。数据预处理两种方法:1.可以不分词直接投入模型;2.可以先将中文文本进行分词。两种方法产生的词汇会非常不同。在后面会具体给出示范。importjiebaimportrefromsklearn.feature_extraction.textimportCountVectorizer#原始数据text=['很少在公众场合手机
本文目录
简介
CountVectorizer官方文档。
将一个文档集合向量化为为一个计数矩阵。
如果不提供一个先验字典,不使用分析器做某种特征选择,那么特征的数量将等于通过分析数据发现的词汇量。
数据预处理
两种方法:1.可以不分词直接投入模型;2.可以先将中文文本进行分词。
两种方法产生的词汇会非常不同。在后面会具体给出示范。
import jieba
import re
from sklearn.feature_extraction.text import CountVectorizer
#原始数据
text = ['很少在公众场合手机外放',
'大部分人都还是很认真去学习的',
'他们会用行动来',
'无论你现在有多颓废,振作起来',
'只需要一点点地改变',
'你的外在和内在都能焕然一新']
#提取中文
text = [' '.join(re.findall('[\u4e00-\u9fa5]+',tt,re.S)) for tt in text]
#分词
text = [' '.join(jieba.lcut(tt)) for tt in text]
text
构建模型
训练模型
#构建模型 vectorizer = CountVectorizer() #训练模型 X = vectorizer.fit_transform(text)
所有词汇:model.get_feature_names()
#所有文档汇集后生成的词汇 feature_names = vectorizer.get_feature_names() print(feature_names)
不分词生成的词汇

分词后生成的词汇

计数矩阵:X.toarray()
#每个文档相对词汇量出现次数形成的矩阵 matrix = X.toarray() print(matrix)

#计数矩阵转化为DataFrame df = pd.DataFrame(matrix, columns=feature_names) df

词汇索引:model.vocabulary_
print(vectorizer.vocabulary_)

作者最新文章
腾讯会议如何删除云录制与本地录像文件
2026-11-09 16:51
xshell连接服务器怎么连 SSH远程登录步骤
2026-11-03 10:13
立刻MV如何删除不需要的MV项目工程
2026-11-02 23:37
beyond compare怎么自动换行
2026-10-23 10:35
Luminar Neo画笔与蒙版使用指南:精准选区与图层合成技巧
2026-10-22 15:23
热门文章
更多
精品专题
更多
Mac软件
更多
火山引擎 网页版
Windows/ macOS/ Linux
火山引擎是字节跳动旗下企业级云与AI服务平台。聚焦大模型核心能力,依托字节自研豆包多模态大模型,凭借火山方舟一站式平台,提供模型精调、推理部署、智能应用搭建等全链路服务,结合高性能AI算力,助力各行业快速落地大模型数字化与智能化应用。
















