sklearn之朴素贝叶斯进行分类

参考：https://www.cnblogs.com/youngsea/p/9327972.html

from sklearn import preprocessing
import numpy as np
import pandas as pd
from sklearn.metrics import classification_report
from sklearn.model_selection import train_test_split
from sklearn.naive_bayes import MultinomialNB
from sklearn.preprocessing import LabelEncoder

'''
使用朴素贝叶斯对adult数据集进行归一化，选取其中选出对收入影响较大的七个特征属性，包括：Age、Workclass、Education、Occupation、race、Sex、Native_country，以这七个属性做为特征属性
'''

col_names = ['age', 'workclass', 'education', 'occupation', 'race', 'sex', 'native-country', 'income']

data = pd.read_csv('D:\\python_project\\classifier-master\\adult.csv', header=None, names=col_names)
col_numbers = 'age'

#先把数据集中的字符串转换成数字
for col in col_names:
    if col not in col_numbers:
        data[col] = LabelEncoder().fit_transform(data[col])

x, y = data[col_names], data['income']
print(x)

#归一化
min_max_scaler = preprocessing.MinMaxScaler()
X_minMax = min_max_scaler.fit_transform(x)
print(X_minMax)

x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.1, random_state=0)

mnb = MultinomialNB()   # 使用默认配置初始化朴素贝叶斯
mnb.fit(x_train,y_train)    # 利用训练数据对模型参数进行估计
y_predict = mnb.predict(x_test)     # 对参数进行预测

#获取结果报告
print('朴素贝叶斯准确率为', mnb.score(x_test,y_test))
print(classification_report(y_test, y_predict))

结果如下：

补充：
x矩阵：

归一化后的：x_minMax矩阵