1. 首页
  2. 数据库
  3. 其它
  4. 子类API实现wide&deep模型——网络层构建

子类API实现wide&deep模型——网络层构建

上传者: 2021-01-16 07:27:01上传 PDF文件 23.65KB 热度 11次
子类API实现wide&deep模型——网络层构建 以下代码展示如何用子类API实现wide&deep模型的网络层的构建,数据是使用的sklearn中的加州房价预测数据集。wide以及deep模型使用同样的输出。 代码展示: class WideDeepModel(keras.models.Model): def __init__(self): super(WideDeepModel, self).__init__() #定义模型层次 self.hidden1_layer = keras.layers.Dense(30, activati
用户评论