从零开始训练神经网络---Keras【学习笔记】[1/2]

WarnerJDoe

已于 2022-06-07 13:50:55 修改

阅读量330

点赞数

分类专栏： AI相关软件学习文章标签： keras 神经网络学习 tensorflow

于 2022-06-05 15:03:15 首次发布

本文链接：https://blog.csdn.net/WarnerJDoe/article/details/125131825

版权

Keras简介：

Keras是由纯python编写的基于theano/tensorflow的深度学习框架。
Keras是一个高层神经网络API，支持快速实验，能够把你的idea迅速转换为结果，如果有如下需求，可以优先选择Keras：

简易和快速的原型设计（keras具有高度模块化，极简，和可扩充特性）
支持CNN和RNN，或二者的结合
无缝CPU和GPU切换

Keras官网首页

Keras官网首页logo

一、背景

本次构建神经网络最终目的：输入一张手写数字图片后，网络输出该图片对应的数字。

二、Keras代码实现

2.1 导入Keras库

import tensorflow as tf
from tensorflow.keras.datasets import mnist

如果没有安装TensorFlow，那么可以参考我之前的文章：重装CUDA和cuDNN(目的是装TensorFlow和pytorch)【个人梳理总结】

2.2 加载数据集(训练集和测试集)

我们使用Keras下的MNIST手写字符数据集，可以使用如下命令下载数据集：

(train_images, train_labels), (test_images, test_labels) = keras.datasets.mnist.load_data()
'''
load_data() returns tuple of NumPy arrays: (x_train, y_train), (x_test, y_test).
train_images: uint8 NumPy array of grayscale image data with shapes (60000, 28, 28), containing the training data. 
Pixel values range from 0 to 255.
train_labels: uint8 NumPy array of digit labels (integers in range 0-9) with shape (60000,) for the training data.
test_images: uint8 NumPy array of grayscale image data with shapes (10000, 28, 28), containing the test data. 
Pixel values range from 0 to 255.
test_labels: uint8 NumPy array of digit labels (integers in range 0-9) with shape (10000,) for the test data.
'''

下面是选做步骤~

如果想验证一下各个变量的shape可以使用python中的断言关键字：

assert train_images.shape == (60000, 28, 28)
assert train_labels.shape == (10000, 28, 28)
assert test_images.shape == (60000,)
assert test_labels.shape == (10000