通过SASRec算法进行基于Transformer的商品推荐

案例简介

(1)方法概述: 本教程包含如下内容：

• 从原始的数据文件中加载数据，进行训练集和测试集的切分，并对测试集进行负采样。

• 对数据分batch, 利用用户历史点击记录进行模型训练

• 结果展示

(2)宏观流程图

2.准备工作

(1) 对应库的安装(以下是运行环境)

tqdm==4.51.0

numpy==1.19.2

torch==1.7.0

(2) 其他环境

python=3.8

数据集：https://download.csdn.net/download/qq_38735017/87164686

[1]

import argparse
import os
import time

import torch
import numpy as np
from tqdm import tqdm

# 引入必要的数据处理以及评测和日志模块
from datasets.src.Logger import Logger
from datasets.src.utils