Tensorflow框架搭建深度学习神经网络笔记(1)——基于tf.keras高阶API

本文介绍了使用Tensorflow的tf.keras高阶API搭建深度学习神经网络的方法,包括通过Sequential API顺序搭建模型,以及编译、训练、评价模型的步骤。示例中详细讲解了如何进行图片预处理、选择交叉熵损失函数,并提供了one-hot和一般形式的损失计算方法。
摘要由CSDN通过智能技术生成

tf.keras是tensorflow目前最新的一个高阶API,目前tf的官网都是以此为指南来教大家入门tensorflow

这个API的使用和此前需要session不同,用这个API搭建de两种方式:

  1. tf.keras.Sequential:一层一层得顺序搭建一个模型
  2. tf.keras.Model:函数API,可以创建多输入、多输出模型

这里的笔记以tf.keras.Sequential为示例

初始化模型

#导入相应的库
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers
#以mnist数据集为例
import keras.datasets.minist as mnist
(train_image,train_lable),(test_image,test_lable) = mnist.load_data()
#查看数据的shape
train_image.shape
import matplotlib.pyplot as plt
#将图片显示在当前页面
%matplotlib inline 
plt.imshow(train_image[3])
#把图片展平,因为这里我们用tf.keras.Sequential这种方式来建立模型
train_image = train_image.reshape(-1,28*28)
#正式开始建立模型
model = keras.Sequential()

添加层,构建网络

#网络的第一层要给定输入数据的shape
model.add(layers.Conv2D(64,(3,3),strides=(1,1),activation='relu',input_shape=(784,)))
model.add(layers.MaxPooling2D((2,2)))
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值