tensorflow2.0 API学习

tf.keras.Sequential

Sequential是一个方法类,可以帮助我们轻而易举地以堆叠神经网络的方式集成构建一个复杂的神经网络模型,Sequential提供了丰富的方法,利用这些方法可以快速的实现神经网络模型的网络层级集成、神经网络模型编译、神经网络模型训练和保存、神经网络模型的加载和预测。
1、神经网络模型的网络层级集成
构建模型后编译
下面我们尝试构建三个全连接层的网络模型

import tensorflow as tf
model = tf.keras.Sequential()
#使用add方法集成神经网络层级
model.add(tf.kears.layers.Dense(256, activation = 'relu'))
model.add(tf.keras.layers.Dense(256, activation = 'relu'))
model.add(tf.keras.layers.Dense(2, activation = 'softmax'))

2.神经网络模型的编译
需编译后才能够进行训练,通过compile()函数编译

#loss:损失函数
#optimizer:选择优化器
#metrics:模型评价的方法,可选mse,accuracy等
model.compile(loss='sparse_categorical_crossentropy',optimizer=tf.kears.optimizers.Adan(0.01),metrics=[''accuracy])

3.神经网络模型训练和保存。
通过fit()函数训练。fit有较多参数,下面一一例举:
x: 配置训练的输入数据,可以是array或者tensor类型。
y: 配置训练的标注数据(即label),可以是array或者tensor类型。
batch_size: 配置批大小, 默认值是32.
epochs: 配置训练的epcohs的数量(说白了就是迭代训练次数)
verbose: 配置训练过程信息输出的级别,共三个级别,0代表不输出任何信息,1代表以进度条的方式输出训练过程信息,2代表每个epoch输出一条训练过程信息
validation_split: 配置验证集占训练数据集的比例,取值范围为0到1,
validation_data: 配置验证集,如果已经配置validation_split,则可以不用配置高该参数,如果同事配置,那么validation_split会自动失效。
shuffle: 配置是否随机打乱训练集。当配置steps_per_epoch为None时,本参数的配置失效。
initial_epoch: 配置进行fine_tune时,新的训练周期是从指定的epoch开始继续训练。
steps_per_peoch:配置每个epoch训练的步数。
通过**model.save()或者save_weights()**方法来保存并导出训练得到的模型。
4.神经网络模型的加载和预测
加载:

tf.keras.mosels.load_model()#加载模型 参数:模型完整名称
predict()#参数:图片数据,  用来预测
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值