Commit cf5048a

committed

savez knn_data.npz

1 parent 2797421 commit cf5048aCopy full SHA for cf5048a

File tree

+30

-2

lines changed

+30

-2

lines changed

Lines changed: 6 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -43,14 +43,18 @@`
`43`	`43`	`accuracy = correct * 100.0 / result.size`
`44`	`44`	`print('准确率', accuracy) # 准确率91%`
`45`	`45`
`46`		`-#`
	`46`	`+''''''`
`47`	`47`	`# save the data`
`48`		`-np.savez('knn_data.npz', train=train, train_labels=train_labels)`
	`48`	`+np.savez('knn_data.npz', train=train, train_labels=train_labels,test=test,test_labels=test_labels)`
`49`	`49`	`# Now load the data`
`50`	`50`	`with np.load('knn_data.npz') as data:`
`51`	`51`	`print(data.files)`
`52`	`52`	`train = data['train']`
`53`	`53`	`train_labels = data['train_labels']`
	`54`	`+ test = data['test']`
	`55`	`+ test_labels = data['test_labels']`
`54`	`56`
`55`	`57`
`56`	`58`	`#TODO 怎样预测数字?`
	`59`	`+# knn.predict?`
	`60`	`+# Docstring: predict(samples[, results[, flags]]) -> retval, results`

Lines changed: 24 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,24 @@`
	`1`	`+# -- coding: utf-8 --`
	`2`	`+# @Time : 2017年8月8日 11:57`
	`3`	`+# @Author : play4fun`
	`4`	`+# @File : 预测手写数字1.py`
	`5`	`+# @Software: PyCharm`
	`6`	`+`
	`7`	`+"""`
	`8`	`+预测手写数字1.py:`
	`9`	`+"""`
	`10`	`+`
	`11`	`+import numpy as np`
	`12`	`+import cv2`
	`13`	`+from matplotlib import pyplot as plt`
	`14`	`+`
	`15`	`+with np.load('knn_data.npz') as data:`
	`16`	`+ print(data.files)`
	`17`	`+ train = data['train']`
	`18`	`+ train_labels = data['train_labels']`
	`19`	`+ test = data['test']`
	`20`	`+ test_labels = data['test_labels']`
	`21`	`+`
	`22`	`+knn = cv2.ml.KNearest_create()`
	`23`	`+knn.train(train, cv2.ml.ROW_SAMPLE, train_labels)`
	`24`	`+ret, result, neighbours, dist = knn.findNearest(test, k=5)`

Comments

(0)