机器学习:如何简单理解【贝叶斯分类】问题
场景
有10个人在会议室开会,其中4名高管,6名普通员工。4名高管中有3人穿衬衫打领带,6名普通员工中有2人穿衬衫打领带。现在推门进来1人,穿衬衫打领带,那么进来的人是高管还是普通员工?
解答
根据朴素贝叶斯公式:
-
第一步:
- A代表是否为高管
- B代表是否穿衬衫打领带
-
第二步:
- 高管中穿衬衫打领带的概率P(B|A)=3/4
- 会议室中高管的概率P(A)=4/10
- 会议室中穿衬衫打领带概率P(B)=5/10
-
第三步:
代入朴素贝叶斯公式,则穿衬衫打领带的人中是高管的概率P(A|B)=(3/4 * 4/10)/(5/10) = 3/5
-
第四步:
参照第一、二、三步骤,则穿衬衫打领带的人中是普通员工的概率P(A|B)=(2/6 * 6/10)/(5/10)=2/5
结论
- 进来的人是高管的概率是3/5
- 进来的人是员工的概率是2/5