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

复制代码
导入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)
复制代码

 

posted @   gugubeng  阅读(293)  评论(0编辑  收藏  举报
努力加载评论中...
点击右上角即可分享
微信分享提示