Tensorflow2.0 利用API 自定义构建回归模型

最新推荐文章于 2024-02-28 20:24:17 发布

付

最新推荐文章于 2024-02-28 20:24:17 发布

阅读量397

点赞数

分类专栏： ML&DL

本文链接：https://blog.csdn.net/F229338596/article/details/107743073

版权

ML&DL 专栏收录该内容

12 篇文章 0 订阅

订阅专栏

低阶API构建回归模型

准备数据

import numpy as np 
import pandas as pd
from matplotlib import pyplot as plt 
#样本数量
n = 400
# 生成测试用数据集
X = tf.random.uniform([n,2],minval=-10,maxval=10) 
w0 = tf.constant([[2.0],[-3.0]])
b0 = tf.constant([[3.0]])
Y = X@w0 + b0 + tf.random.normal([n,1],mean = 0.0,stddev= 2.0)  # @表示矩阵乘法,增加正态扰动

数据可视化

# 数据可视化

%matplotlib inline
# %config InlineBackend.figure_format = 'svg'

plt.figure(figsize = (5,5))
ax1 = plt.subplot(121)
ax1.scatter(X[:,0],Y[:,0], c = "b")
plt.xlabel("x1")
plt.ylabel("y",rotation = 0)

ax2 = plt.subplot(122)
ax2.scatter(X[:,1],Y[:,0], c = "g")
plt.xlabel("x2")
plt.ylabel("y",rotation = 0)
plt.show()

在这里插入图片描述

数据管道

### 数据迭代器
### tf.gather 获取指定列表中 指定索引的值
def data_iter(features, labels, batch_size=8):
    num_examples = len(features)
    indices = list(range(num_examples))
    np.random.shuffle(indices)  #样本的读取顺序是随机的
    for i in range(0, num_examples, batch_size):
        ## 防止 上越界
        indexs = indices[i: min(i + batch_size, num_examples)]
        yield tf.gather(features,indexs), tf.gather(labels,indexs)

定义模型

w = tf.Variable(tf.random.normal(w0.shape))
b = tf.Variable(tf.zeros_like(b0,dtype = tf.float32))

# 自定义模型
class LinearRegression:     
    #正向传播
    def __call__(self,x): 
        return x@w + b ## @表示矩阵乘法
    # 损失函数
    def loss_func(self,y_true,y_pred):  
        return tf.reduce_mean((y_true - y_pred)**2/2)

model = LinearRegression()

训练，旨在更新参数，获取对应的损失及其他输出

##使用autograph机制转换成静态图加速

@tf.function
def train_step(model, features, labels):
    with tf.GradientTape() as tape:
        predictions = model(features)
        loss = model.loss_func(labels, predictions)
    # 反向传播 求梯度（线性函数其实没有反向传播，可以直接求各个参数的倒数）
    dloss_dw, dloss_db = tape.gradient(loss,[w,b])
    # 梯度下降法 更新参数
    w.assign(w - 0.001*dloss_dw)
    b.assign(b - 0.001*dloss_db)
    
    return loss
    
def train_model(model,epochs):
    for epoch in tf.range(1,epochs+1):
        for features, labels in data_iter(X,Y,10):
            loss = train_step(model,features,labels)
        if epoch%50==0:
            ##printbar()
            tf.print("epoch =",epoch,"loss = ",loss)
            tf.print("w =",w)
            tf.print("b =",b)

train_model(model,epochs = 200)
## 以下展示的是学习出来的 某一维度 与 label的关系图

在这里插入图片描述

高阶API构建回归模型

此处值列举与原生API 差异。

构建模型差异。高阶api可以直接定义模型，添加Layer ，这样就基本完成网络的构建

tf.keras.backend.clear_session()
model = models.Sequential()
model.add(layers.Dense(1,input_shape =(2,)))
model.summary()

训练差异。

### 使用fit方法进行训练
#### 可以指明损失函数，优化器，评估函数
model.compile(optimizer="adam",loss="mse",metrics=["mae"])
#### 定义训练过程中参数，迭代次数，每次feed 多少数据 batch_size
model.fit(X,Y,batch_size = 10,epochs = 200)  
tf.print("w = ",model.layers[0].kernel)
tf.print("b = ",model.layers[0].bias)