keras 多输入模型

def build_model(product_shape, level_shape, attr_shape, period_shape):
    product_inputs = keras.Input(shape=(product_shape, ))
    level_inputs = keras.Input(shape=(level_shape, ))
    attr_inputs = keras.Input(shape=(attr_shape, ))
    period_inputs = keras.Input(shape=(period_shape, ))

    product_dense = keras.layers.Dense(256, activation='relu')(product_inputs)
    product_dense= keras.layers.BatchNormalization()( product_dense)
    

    laptop_inputs = keras.layers.concatenate([product_dense, level_inputs, attr_inputs, period_inputs])
    laptop_dense = keras.layers.Dense(256, activation='relu')(laptop_inputs)
    laptop_dense = keras.layers.BatchNormalization()( laptop_dense)
    laptop_dense = keras.layers.Dense(128, activation='relu')(laptop_dense)
    laptop_dense = keras.layers.BatchNormalization()( laptop_dense)
    laptop_dense = keras.layers.Dense(64, activation='relu')(laptop_dense)
    laptop_dense = keras.layers.BatchNormalization()( laptop_dense)
    outputs = keras.layers.Dense(1, activation='linear')(laptop_dense)



    model = keras.Model(inputs=[product_inputs, level_inputs, attr_inputs, period_inputs], outputs=outputs)
    opt = keras.optimizers.Adam()
#    opt = keras.optimizers.RMSprop(lr=3e-3)
    model.compile(optimizer=opt, loss='mse')
    
    return model

posted @   luoganttcc  阅读(12)  评论(0编辑  收藏  举报
编辑推荐:
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
· AI与.NET技术实操系列:向量存储与相似性搜索在 .NET 中的实现
· 基于Microsoft.Extensions.AI核心库实现RAG应用
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
阅读排行:
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· Manus的开源复刻OpenManus初探
· 写一个简单的SQL生成工具
· AI 智能体引爆开源社区「GitHub 热点速览」
· C#/.NET/.NET Core技术前沿周刊 | 第 29 期(2025年3.1-3.9)
点击右上角即可分享
微信分享提示