码迷,mamicode.com
首页 > 其他好文 > 详细

机器学习(三) 贝叶斯文本分类问题

时间:2018-10-17 14:39:09      阅读:114      评论:0      收藏:0      [点我收藏+]

标签:import   ssi   导入   form   targe   ext   rac   target   div   

导入sklearn.feature_extraction.text.TfidfVectorizer用于转换字符串


from sklearn.feature_extraction.text import TfidfVectorizer

# 创建对象
tf = TfidfVectorizer()
# 训练
tf.fit(data)
# transform 转化
data_result = tf.transform(data).toarray()

ga_NB = GaussianNB()
ga_NB.fit(data_result, target)

 

机器学习(三) 贝叶斯文本分类问题

标签:import   ssi   导入   form   targe   ext   rac   target   div   

原文地址:https://www.cnblogs.com/gugubeng/p/9803458.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!