Tensorflow函数式API的使用

在我们使用tensorflow时,如果不能使用函数式api进行编程,那么一些复杂的神经网络结构就不会实现出来,只能使用简单的单向模型进行一层一层地堆叠。如果稍微复杂一点,遇到了Resnet这种带有残差模块的神经网络,那么用简单的神经网络堆叠的方式则不可能把这种网络堆叠出来。下面我们来使用函数式API来编写一个简单的全连接神经网络:
首先导包:

from tensorflow import keras
import tensorflow as tf
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

导入图片数据集:mnist

(train_image,train_label),(test_image,test_label)=tf.keras.datasets.fashion_mnist.load_data()

归一化:

train_image=train_image/255
test_image=test_image/255#进行数据的归一化,加快计算的进程

搭建全连接神经网络:

input=keras.Input(shape=(28,28))
x=keras.layers.Flatten()(input)#调用input
x=keras.layers.Dense(32,activation="relu")(x)
x=keras.layers.Dropout(0.5)(x)#一层一层的进行调用上一层的结果
output=keras.layers.Dense(10,activation="softmax")(x)
model=keras.Model(inputs=input,outputs=output)
model.summary()

输出:

Model: "model"
_________________________________________________________________
Layer (type)            
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值