首页 > 编程语言 > 详细

机器学习算法的分类准确度

时间:2019-04-07 11:58:44      阅读:147      评论:0      收藏:0      [点我收藏+]

本章,我们使用sklearn自带的手写识别的数据集进行计算准确度,进而巩固之前学的KNN算法。

import numpy as np
import matplotlib.pyplot as plt
from sklearn import datasets
from sklearn.neighbors import KNeighborsClassifier
from sklearn.model_selection import train_test_split


digits = datasets.load_digits()

x = digits.data #获取特征值
y = digits.target #获取标记

#将数据分为两部分,训练数据和测试数据

x_train,x_test,y_train,y_test = train_test_split(x,y,test_size=0.2)

#指定key值
knn = KNeighborsClassifier(n_neighbors=3)

#进行拟合
knn.fit(x_train,y_train)

y_predict = knn.predict(x_test)

ratio = sum(y_predict==y_test)/len(y_test)

print(ratio)

#当我们不想要预测值的时候,我们可以直接使用knn对象的score函数进行得出准确度
ratio_bak = knn.score(x_test,y_test)
print(ratio_bak)

本节主要是进行之前学的KNN算法进行巩固,进而为后续的学习打好基础。

机器学习算法的分类准确度

原文:https://www.cnblogs.com/lyr999736/p/10664203.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!