大师兄的数据分析学习笔记(十六):分类模型(二)

2022-07-08  本文已影响0人  superkmi

大师兄的数据分析学习笔记(十五):分类模型(一)
大师兄的数据分析学习笔记(十七):分类模型(三)

二、朴素贝叶斯

1. 回顾概率
  • 贝叶斯公式用来描述两个条件概率之间的关系。
  • 从上面的公式可以看出,贝叶斯公式的分子可以看作联合概率,而分母可以看作全概率
2. 朴素贝叶斯的基本思想
  • F_1:日志数量/注册天数,离散化:F_1<=0.05, 0.05<F_1<0.2, F_1>=0.2
  • F_2:好友数量/注册天数,离散化:F_2<=0.1, 0.1<F_2<0.8, F_2>0.8
  • F_3:是否使用真实头像,离散化: True = 1/False = 0

真实账号落入F_1的比例为0.5:P(F1=1|C_0)=0.5 / 虚假账号落入F_1的比例为0.1:P(F_1|C_1)=0.1
真实账号落入F_2的比例为0.7:P(F1=1|C_0)=0.7 / 虚假账号落入F_1的比例为0.2:P(F_1|C_1)=0.2
真实账号落入F_3的比例为0.2:P(F1=1|C_0)=0.2 / 虚假账号落入F_1的比例为0.2:P(F_1|C_1)=0.9

P(F_1=1|C_0)P(F_2=1|C_0)P(F_3=0|C_0)P(C_0) = 0.5 \times 0.7 \times 0.2 \times 0.89 = 0.0623
P(F_1=1|C_1)P(F_2=1|C_1)P(F_3=0|C_1)P(C_1) = 0.1 \times 0.2 \times 0.9 \times 0.11 = 0.00198

3. 拉普拉斯平滑
4. 代码实现
>>>import os
>>>import pandas as pd
>>>import numpy as np
>>>from sklearn.model_selection import train_test_split
>>>from sklearn.naive_bayes import GaussianNB,BernoulliNB
>>>from sklearn.metrics import  accuracy_score,recall_score,f1_score

>>>models = []
>>>models.append(("GaussianNB",GaussianNB()))
>>>models.append(("BernoulliNB",BernoulliNB()))

>>>df = pd.read_csv(os.path.join(".", "data", "WA_Fn-UseC_-HR-Employee-Attrition.csv"))
>>>X_tt,X_validation,Y_tt,Y_validation = train_test_split(df.JobLevel,df.JobSatisfaction,test_size=0.2)
>>>X_train,X_test,Y_train,Y_test = train_test_split(X_tt,Y_tt,test_size=0.25)

>>>data = df[["JobSatisfaction","JobLevel"]]

>>>for clf_name,clf in models:
>>>    clf.fit(np.array(X_train).reshape(-1,1),np.array(Y_train).reshape(-1,1))
>>>    xy_lst = [(X_train,Y_train),(X_validation,Y_validation),(X_test,Y_test)]
>>>    for i in range(len(xy_lst)):
>>>        X_part = xy_lst[i][0]
>>>        Y_part = xy_lst[i][1]
>>>        Y_pred = clf.predict(np.array(X_part).reshape(-1,1))
>>>        print(i)
>>>        print(clf_name,"-ACC",accuracy_score(Y_part,Y_pred))
>>>        print(clf_name,"-REC",recall_score(Y_part,Y_pred,average='macro'))
>>>        print(clf_name,"-F1",f1_score(Y_part,Y_pred,average='macro'))
>>>       print("="*40)
0
GaussianNB -ACC 0.3253968253968254
GaussianNB -REC 0.25
GaussianNB -F1 0.12275449101796408
========================================
1
GaussianNB -ACC 0.2755102040816326
GaussianNB -REC 0.25
GaussianNB -F1 0.10799999999999998
========================================
2
GaussianNB -ACC 0.30952380952380953
GaussianNB -REC 0.25
GaussianNB -F1 0.11818181818181818
========================================
0
BernoulliNB -ACC 0.3253968253968254
BernoulliNB -REC 0.25
BernoulliNB -F1 0.12275449101796408
========================================
1
BernoulliNB -ACC 0.2755102040816326
BernoulliNB -REC 0.25
BernoulliNB -F1 0.10799999999999998
========================================
2
BernoulliNB -ACC 0.30952380952380953
BernoulliNB -REC 0.25
BernoulliNB -F1 0.11818181818181818
========================================
5. 生成模型与判别模型
上一篇下一篇

猜你喜欢

热点阅读