当前位置：网站首页>Tensorflow模型整体构建流程

Tensorflow模型整体构建流程

2022-08-09 21:35:00 【phac123】

文章目录

TF构造八股文
简单实现

TF构造八股文

Import: 导入相关的包
train and test: 处理好训练与测试数据集,包括乱序等操作
models.Sequential: 搭建网络模型
model.compile: 设置优化算法,配置训练方法
model.fit: 设置训练时的学习率,迭代次数,batch_size,对模型进行训练;
model.summary：打印出网络结构和参数统计

简单实现

# 1. import
import tensorflow as tf

# 2. train and test
w = tf.Variable(tf.constant(5, dtype=tf.float32))

# 3. model.sequential

# 4. model.compile and 5. model.fit
lr = 0.2
epochs = 40

for epoch in range(epochs):
    with tf.GradientTape() as tape:
        loss = tf.square(w + 1)        # 定义loss函数
    grads = tape.gradient(loss, w)     # 求loss对w的梯度

    w.assign_sub(lr * grads)
    print("After %s epoch, w is %f, loss is %f" % (epoch, w.numpy(), loss))     # 6. model.summary()