代码，逻辑回归(logistic_regression)实现mnist分类（TensorFlow实现）

SusanZhang1231

于 2017-08-07 12:04:39 发布

阅读量1.2k

点赞数

CC 4.0 BY-SA版权

文章标签：逻辑回归Python TensorFlow 代码

本文链接：https://blog.youkuaiyun.com/SusanZhang1231/article/details/76830596

本文通过TensorFlow库详细讲解如何利用逻辑回归算法对MNIST手写数字数据集进行分类，随着epoch参数的调整，模型的分类效果会有所变化。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#logistic_regression by ffzhang
import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
os.environ["CUDA_VISIBLE_DEVICES"]='2'

import numpy as np
import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
import time

mnist=input_data.read_data_sets('data/mnist',one_hot=True)

mnist.train.images.shape

mnist.train.labels.shape

batch_size=128
# X = tf.placeholder