机器学习-手写数字识别1

 用到的库：

        from sklearn import datasets

        from sklearn.model_selection import train_test_split

        from sklearn import svm

        from sklearn.externals import joblib

1、数据的采集与标记 

from sklearn import datasets
digits=datasets.load_digits()#加载sklearn自带的数据集

images_and_labels=list(zip(digits.images,digits.target))#1行若干列
#zip:返回一个以元组为元素的列表 第一个参数的第一个数和第二个参数的第一个数是一组

#显示数据图片
import matplotlib.pyplot as plt
plt.plot(figsize=(8,8),dpi=200)
for index,(image,label) in enumerate(images_and_labels[:8]):#enumerate枚举对象 前8列
    plt.subplot(2,4,index+1)#子图：子图的行、列、当前激活的子图序号
    plt.axis('off')#关闭坐标轴
    plt.imshow(image,cmap=plt.cm.gray_r,interpolation='nearest')
    #plt.imshow()函数负责对图像进行处理，并显示其格式。
    #cmap：将数据映射到色彩图
    #,interpolation='nearest'：插值方法
    plt.title('Digit:%i'%label,fontsize=20)

 2、特征的选择

 3、模型训练、模型测试、模型预测

#把数据分成训练集和测试集
from sklearn.model_selection import train_test_split
Xtrain,Xtest,Ytrain,Ytest=train_test_split(digits.data,digits.target,test_size=0.20,random_state=2)
#random_state 设置随机种子，只要random_state设置相同，结果也完全一样
#使用向量机来训练模型
from sklearn import svm
clf=svm.SVC(gamma=0.001,C=100.)
clf.fit(Xtrain,Ytrain)#训练完成
clf.score(Xtest,Ytest)#评估模型准确度
Ypred=clf.predict(Xtest)
#查看预测情况
fig,axes=plt.subplots(4,4,figsize=(8,8))
fig.subplots_adjust(hspace=0.1,wspace=0.1)
for i,ax in enumerate(axes.flat):#遍历每个坐标轴
    ax.imshow(Xtest[i].reshape(8,8),cmap=plt.cm.gray_r,interpolation='nearest')
    ax.text(0.05,0.05,str(Ypred[i]),fontsize=32,transform=ax.transAxes,
            color='green' if Ypred[i]==Ytest[i] else 'red')#相等是绿色，否则是红色
    #transform=ax.transAxes:文本的位置是以坐标轴为参照物，默认是坐标点
    ax.text(0.8,0.05,str(Ytest[i]),fontsize=32,transform=ax.transAxes,
            color='black')
    ax.set_xticks([])#设置x轴刻度
    ax.set_yticks([])#设置y轴刻度

 

 4、模型的保存和加载

#保存模型参数.这样下次需要预测时，可以直接模型进行 ，而不是重新遍模型。
from sklearn.externals import joblib
joblib.dump(clf,'digits_svm.pkl')

#加载模型
clf_new=joblib.load('digits_svm.pkl')
ypred_new=clf_new.predict(Xtest)
clf_new.score(Ytest,ypred_new)