朴素贝叶斯（Naive Bayes）算法 - 文档分类（敏感词判断）

最新推荐文章于 2025-04-10 08:00:00 发布

原创

最新推荐文章于 2025-04-10 08:00:00 发布 · 1.2k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#机器学习 #朴素贝叶斯 #文档分类 #敏感词

import numpy as np
import pandas as pd
from functools import reduce

'''
	函数功能：创建实验数据集
	参数说明：无参数
	返回：
		postingList：切分好的样本词条
		classVec：类标签向量
'''
def loadDataSet():

	dataSet = [
		['my', 'dog', 'has', 'flea', 'problems', 'help', 'please'],
		['maybe', 'not', 'take', 'him', 'to', 'dog', 'park', 'stupid'],
		['my', 'dalmation', 'is', 'so', 'cute', 'I', 'love', 'him'],
		['stop', 'posting', 'stupid', 'worthless', 'garbage'],
		['mr', 'licks', 'ate', 'my', 'steak', 'how', 'to',