Skip to content

A two tower recommendation system implementation with tensorflow estimator, for CTR or Recall.

Notifications You must be signed in to change notification settings

cdj0311/two_tower_recommendation_system

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

two_tower_recommendation_system

A two tower recommendation system implementation with tensorflow estimator.

基于tensorflow estimator API实现的双塔DNN推荐算法,可作为推荐算法模板在此基础上根据需求修改。

输入特征有:

向量类特征: user向量、item向量

分桶类特征: 年龄

hash类特征: deviceID、itemID

直接使用train_local.sh即可在本地训练,如果需要分布式训练,需设置train_on_cluster=True,然后提交到job中,由于每个公司的job提交命令不一样,这里就不贴出了。

About

A two tower recommendation system implementation with tensorflow estimator, for CTR or Recall.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published