参考书籍

第三章内容

digit 手写数据库

# 最初的分类器
# digits 手写数字库

import numpy as np
import matplotlib.pyplot as plt
from sklearn import datasets
from sklearn import tree
# 性能评价相关的库
from sklearn import metrics

# digits 数据加载
digits = datasets.load_digits()

# 显示此库中第一行的数据
data = digits.data
print(data[0])

# 将画像以2行5列表示
# 前10个手写数字表示
for label, img in zip(digits.target[:10], digits.images[:10]):
    plt.subplot(2,5, label+1)
    plt.axis('off')
    plt.imshow(img, cmap=plt.cm.gray_r, interpolation='nearest')
    plt.title('Digit: {0}'.format(label))

plt.show()

# 求 3 和 8 的位置
flag_3_8 = (digits.target == 3) + (digits.target == 8)

# 获取 3 和 8 的数据
images = digits.images[flag_3_8]
labels = digits.target[flag_3_8]

# 将 3 和 8 的画像数据一次元化
images = images.reshape(images.shape[0], -1)

# 生成分类器
n_samples = len(flag_3_8[flag_3_8])
train_size = int(n_samples * 3 / 5)
# 在tree.DecisionTreeClassifier中生成分类器
classifier = tree.DecisionTreeClassifier()
# 将学习数据给到classifier.fit
# 为了简单使用了60% 的数据作为训练数据
# train_size是作为训练和测试的分界点
classifier.fit(images[:train_size], labels[:train_size])



# 使用剩下的数据进行测试，也就是40%
# 因此这里的测试数据是从 train_size 开始的
# 提取测试数据的标签作为正确标签。
expected = labels[train_size:]
# 分类执行，取出测试数据的预测结果
predicted = classifier.predict(images[train_size:])

# 打印结果
# accuracy_score用来计算正确率
print('accuracy: \n', metrics.accuracy_score(expected, predicted))

结果

[ 0. 0. 5. 13. 9. 1. 0. 0. 0. 0. 13. 15. 10. 15. 5. 0. 0. 3.
15. 2. 0. 11. 8. 0. 0. 4. 12. 0. 0. 8. 8. 0. 0. 5. 8. 0.
0. 9. 8. 0. 0. 4. 11. 0. 1. 12. 7. 0. 0. 2. 14. 5. 10. 12.
0. 0. 0. 0. 6. 13. 10. 0. 0. 0.]
accuracy:
0.8531468531468531