案例简介
(1)方法概述: 本教程包含如下内容:
-
从原始的数据文件中加载数据,进行训练集和测试集的切分,并对测试集进行负采样。
-
对数据分batch, 利用用户历史点击记录进行模型训练
-
结果展示
(2)宏观流程图

2.准备工作
(1) 对应库的安装(以下是运行环境)
tqdm==4.51.0
numpy==1.19.2
torch==1.7.0
(2) 其他环境
python=3.8
数据集:https://download.youkuaiyun.com/download/qq_38735017/87164686
[1]
import argparse
import os
imp
本教程详细介绍了如何利用SASRec Transformer算法进行商品推荐。内容包括数据加载与预处理、模型构建及优化思路。模型基于用户历史点击记录,采用Transformer结构,利用多头注意力机制进行训练,并探讨了self-attention和多头注意力的实现细节。优化方向包括反馈层替代、增强模型以及loss函数调整。
订阅专栏 解锁全文
1557

被折叠的 条评论
为什么被折叠?



