solidglue / DNN_for_YouTube_Recommendations

YouTube推荐系统深度学习召回排序算法, Deep Neural Networks for YouTube Recommendations. YouTubeDNN.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Deep Neural Networks for YouTube Recommendations

以"DNN_for_YouTube_Recommendations"模型和电影评分数据集(ml-1m)为基础,详尽的展示了如何基于TensorFlow2实现推荐系统排序模型。

Introduce

本项目只实现了论文中的排序模型,召回模型同理。项目核心代码都在".py"文件中,所有".ipynb"中的代码只用作演示和单元测试,独立于业务。项目做了一些扩展,例如不定长多值序列特征的embedding(例如用户评分电影序列),简单多目标学习(例如预测评分分数和评分分类)。
This project only implements the ranking model in the paper for demonstration, and the recall model is the same.All core codes are in the ".py" files, and the ".ipynb" scripts are only used for Unit Test and demonstration purposes.Easy to join Attention in the future, this project has made some extensions on the ranking model in the paper, such as Multi values Embedding (such as "rated_movies_lastN" feature) and Simple Multi-Task learning.

Paper:https://dl.acm.org/doi/abs/10.1145/2959100.2959190
DataSet: ml-1m(movie rating)

Notice

如果通过Github站内超链接打开Jupyter Notebook文件发生错误,可以点击根据 https://nbviewer.org 生成的“备用链接”间接访问对应文件。
或者通过以下链接访问整个项目的站外备用链接,注意点击站外备用链接里的非Jupyter Notebook格式文件会跳转回到Github仓库内:
DNN_for_YouTube_Recommendations

Model & Model Trainning

加载数据(Load Datasets)     (备用链接)
构造样本(Create Samples)     (备用链接)
模型定义(Modle Structure)     (备用链接)
模型训练(Modle Trainning)     (备用链接)
保存模型与签名(Save Modles)     (备用链接)
模型推理(Modle Inference)     (备用链接)
main.py

Ranking model in the paper

Deep ranking network architecture
alt text

Ranking model in this project

Simple Multi-Task learning ranking network architecture
alt text

*扩展

1.推荐系统
王树森推荐系统公开课 - 基于小红书的场景讲解工业界真实的推荐系统。
Recommender_System

2.推荐系统推理服务
基于Goalng、Docker和微服务**实现了高并发、高性能和高可用的推荐系统推理微服务,包括多种召回/排序服务,并提供多种接口访问方式(REST、gRPC和Dubbo)等,每日可处理上千万次推理请求。
推荐系统推理微服务Golang

3.机器学习 Sklearn入门教程
机器学习Sklearn入门教程

4.深度学习TensorFlow入门教程
深度学习TensorFlow入门教程

About

YouTube推荐系统深度学习召回排序算法, Deep Neural Networks for YouTube Recommendations. YouTubeDNN.

License:Apache License 2.0


Languages

Language:Jupyter Notebook 92.2%Language:Python 7.8%