[deeplearning-021] tf的最近邻算法

1.从tensorflow的example改编

2.源码

import numpy as np
import tensorflow as tf

from tensorflow.examples.tutorials.mnist import input_data

#one_host = True，把类标记转换成向量，比如类标记是3，就转换成[0,0,0,1,0...0]
mnist = input_data.read_data_sets('/tmp/data', one_hot=True)

#训练集和测试集
Xtrain, Ytrain = mnist.train.next_batch(5000)
Xtest, Ytest = mnist.test.next_batch(200)

#建图
xtrain = tf.placeholder('float', [None, 784])
xtest = tf.placeholder('float', [784])

distance = tf.reduce_sum(tf.abs(xtrain-xtest), reduction_indices=1)
pred = tf.argmin(distance, 0)

with tf.Session() as session:
    accuracy = 0
    for i in range(len(Xtest)):
        # dis = session.run(distance, feed_dict={xtrain:Xtrain, xtest:tt})
        p = session.run(pred, feed_dict={xtrain:Xtrain, xtest:Xtest[i]})
        if np.argmax(Ytest[i]) == np.argmax(Ytrain[p]):
            accuracy += 1./len(Xtest)

    print(accuracy)