08-Tensorflow2.x Wide&Deep模型

最新推荐文章于 2023-06-03 02:40:43 发布

AIChemistr

最新推荐文章于 2023-06-03 02:40:43 发布

阅读量389

点赞数

分类专栏： Tensorflow2.0课程文章标签： tensorflow 神经网络 python

本文链接：https://blog.youkuaiyun.com/qq_44783177/article/details/108106548

版权

文章目录

Wide & Deep model简要介绍

在这里插入图片描述

模型的主要程序，在model建立中介绍如下：

1.函数式API ，功能API

# model建立部分
# 函数式API 功能API
input = keras.layers.Input(shape=x_train.shape[1:])
hidden1 = keras.layers.Dense(30,activation='relu')(input)
hidden2 = keras.layers.Dense(30,activation='relu')(hidden1)
# 复合函数：f(x) = h(g(x))
concat = keras.layers.concatenate([input,hidden2])
output = keras.layers.Dense(1)(concat)
model = keras.models.Model(inputs=[input],
                           outputs=[output])

2.子类API

# model建立部分
# 子类API
class WideDeepModel(keras.models.Model):
    def __init__(self):
        super(WideDeepModel,self).__init__()
        """定义模型的层次"""
        self.hidden1_layer = keras.layers.Dense(30,activation='relu')
        self