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

最新推荐文章于 2023-09-19 21:31:57 发布

SusanZhang1231

最新推荐文章于 2023-09-19 21:31:57 发布

阅读量1.2k

点赞数

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

本文链接：https://blog.csdn.net/SusanZhang1231/article/details/76830596

版权

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

摘要由CSDN通过智能技术生成

#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=