繁体   English   中英

为什么我的accuracy_score 指标不正确? scikit 学习

[英]Why is my accuracy_score metric incorrect? scikit learn

我有一些工作代码,这给我带来了麻烦。 我似乎得到了一个几乎随机的accuracy_score 指标,而我的预测值的打印输出表明并非如此。 我正在在线学习教程,这是我到目前为止所写的内容:

import os
import pandas as pd
from sklearn.preprocessing import LabelEncoder
from sklearn.model_selection import train_test_split
from sklearn.naive_bayes import GaussianNB
from sklearn.metrics import accuracy_score, confusion_matrix

adult_train = pd.read_csv(os.path.expanduser("~/Desktop/") + "adult_train_srt.csv", sep=',')
print(adult_train.head(100))

le = LabelEncoder()
adult_train['age'] = le.fit_transform(adult_train['age'])
adult_train['workclass'] = le.fit_transform(adult_train['workclass'].astype(str))
adult_train['education'] = le.fit_transform(adult_train['education'].astype(str))
adult_train['occupation'] = le.fit_transform(adult_train['occupation'].astype(str))
adult_train['race'] = le.fit_transform(adult_train['race'].astype(str))
adult_train['sex'] = le.fit_transform(adult_train['sex'].astype(str))
adult_train['hours_per_week'] = le.fit_transform(adult_train['hours_per_week'])
adult_train['native_country'] = le.fit_transform(adult_train['native_country'].astype(str))
adult_train['classs'] = le.fit_transform(adult_train['classs'].astype(str))

cols = [col for col in adult_train.columns if col not in ['classs']]
data = adult_train[cols]
target = adult_train['classs']

data_train, data_test, target_train, target_test = train_test_split(data, target, test_size = 0.1) #, random_state = 42)

gnb = GaussianNB()
pred = gnb.fit(data_train, target_train).predict(data_test)
pred_gnb = gnb.predict(data_test)
print(pred_gnb)

print("Naive-Bayes accuracy: (TN + TP / ALL) ", accuracy_score(pred_gnb, target_test)) #normalize = True
print("""Confusion matrix:
TN - FP
FN - TP
Guessed:
0s +, 1s -
0s -, 1s +
""")
print(confusion_matrix(target_test, pred_gnb))

Prediction = pd.DataFrame({'Prediction':pred_gnb})

result = pd.concat([adult_train, Prediction], axis=1)
print(result.head(10))

我不知所措,我无法理解我的数据帧连接是否有效,或者accuracy_score 指标是否正在解决其他问题,因为我得到如下输出: 在此处输入图片说明

这个特殊的例子是说有 7 个真阴性(OK)、1 个假阳性(???)、2 个假阴性(OK)和 0 个真阳性(???,但有一个猜对了?)。 [classs] 列是 [Prediction] 列正在猜测的内容。

result = pd.concat([adult_train, Prediction], axis=1)

这里的 Prediction 数据框,不应该和 Adult_train 连接,Prediction 是对测试集 data_set 的预测结果

pred_gnb = gnb.predict(data_test)

所以,我认为你应该连接 data_test、target_test 和 Prediction,试试这个,它可能会工作

result = pd.concat([pd.DataFrame(data_test), pd.DataFrame(target_test), Prediction], axis=1)

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM