tensorflow学习笔记8

逻辑回归框架1

import numpy as np
import tensorflow as tf
import matplotlib.pyplot as plt
import input_data

mnist = input_data.read_data_sets('data/',one_hot=True) #one_hot=True编码格式为01编码
trainimg = mnist.train.images
trainlabel = mnist.train.labels
testimg = mnist.test.images
testlabel = mnist.test.labels

print(trainimg.shape)
print(trainlabel.shape)
print(testimg.shape)
print(testlabel.shape)
print(trainlabel[0])

 

 

trainimg55000个数据,28*28=784个像素

trainlabel10个label(0-9十个数字)

posted @ 2021-01-24 22:33  藻类植物  阅读(24)  评论(0编辑  收藏  举报