Python ScikitLearn 손글씨 숫자 인식 2

 

이전에 학습 내용(https://infos.tistory.com/3578) 을 불러와서 테스트

난수로 선택된 임의의 숫자를 선택하고

숫자에 해당하는 이미지를 넣어 결과 확인

 

from sklearn import datasets, svm, metrics, neighbors, model_selection, metrics
import joblib

import matplotlib.pyplot as plt
import random

clf = joblib.load('digits.pkl')

digits = datasets.load_digits()

for i in range(10):
    plt.subplot(3,5, i+1)
    plt.axis("off")
    n = random.randrange(0, 10)
    plt.title(str(clf.predict(digits.images[n].reshape(-1,64))))
    plt.imshow(digits.images[n])

plt.show()