双塔模型-向量内积-tf.keras.layers.Dot()使用
Dot层计算两个向量中的样本间的点乘,同时也是基于batch的点乘,使用起来非常方便,在模型输出前不用再独立书写cosine部分
返回:shape为(batch_size, 1),表明batch中每一组对应样本的点乘。注意:如果在双塔中使用,在Dot前要对塔两侧向量做l2范数归一化。
示例代码:
def tow_tower(feature_inputs, item_feature_columns, user_feature_columns, hidden_units): # 物品侧输入特征层 item_tower = tf.keras.layers.DenseFeatures(item_feature_columns)(feature_inputs) # 物品塔结构 for num_nodes in hidden_units: item_tower = tf.keras.layers.Dense(num_nodes, activation='relu')(item_tower) item_tower = Lambda(lambda x :K.l2_normalize(x, axis=1))(item_tower) # 用户侧输入特征层 user_tower = tf.keras.layers.DenseFeatures(user_feature_columns)(feature_inputs) # 用户塔结构 for num_nodes in hidden_units: user_tower = tf.keras.layers.Dense(num_nodes, activation='relu')(user_tower) user_tower = Lambda(lambda x :K.l2_normalize(x, axis=1))(user_tower) output = tf.keras.layers.Dot(axes=1)([item_tower, user_tower]) # 定义keras模型 model = tf.keras.Model(feature_inputs, output) return model
时刻记着自己要成为什么样的人!
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享4款.NET开源、免费、实用的商城系统
· 全程不用写代码,我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了
· 上周热点回顾(2.24-3.2)