KNN-mnist数据集识别

最新推荐文章于 2025-05-15 12:40:27 发布

寒江共雪

最新推荐文章于 2025-05-15 12:40:27 发布

阅读量649

点赞数

CC 4.0 BY-SA版权

分类专栏： Python TensorFlow 文章标签： Tensorflow

本文链接：https://blog.csdn.net/CQDIY/article/details/88116681

KNN-mnist数据集识别

win10
python3.6
tensorflow1.12

import numpy as np
import tensorflow as tf

# 加载 mnist 数据
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("/mnist/", one_hot=True)

# 限制数据集数目
X_train, Y_train = mnist.train.next_batch(5000)
X_test, Y_test = mnist.test.next_batch(200)

# 输入训练数据
x_train