标签:images test shape train 报错 tf tensoflow2.1 mnist
新建MNIST_data文件夹,放入下载的四个mnist文件
链接:https://pan.baidu.com/s/13FzOUnvOkvZ8PKFaPXumdw
提取码:1234
目录结构
test.py代码
# -*- coding: utf-8 -*- import tensorflow as tf from tensorflow.examples.tutorials.mnist import input_data #通过包引入类 mnist = input_data.read_data_sets("MNIST_data/", one_hot=True) #编码为one_hot格式 print ('输入数据:',mnist.train.images) print ('输入数据打shape:',mnist.train.images.shape) print ('输入数据打shape:',mnist.test.images.shape) #测试集 print ('输入数据打shape:',mnist.validation.images.shape) #验证集 x_train = mnist.train.images y_train = mnist.train.labels x_test = mnist.test.images y_test = mnist.test.labels x_train, x_test = x_train / 255.0, x_test / 255.0 model = tf.keras.models.Sequential([ tf.keras.layers.Flatten(), tf.keras.layers.Dense(128, activation='relu'), tf.keras.layers.Dense(10, activation='softmax') ]) model.compile(optimizer='adam', loss=tf.keras.losses.CategoricalCrossentropy(from_logits=False), #loss='mean_squared_error', metrics=['categorical_accuracy']) model.fit(x_train, y_train, batch_size=32, epochs=10, validation_data=(x_test, y_test), validation_freq=1) model.summary()
标签:images,test,shape,train,报错,tf,tensoflow2.1,mnist 来源: https://www.cnblogs.com/MAX-ZMY/p/14041380.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。