Skip to content

数据预处理步骤错误 #16

Open

Description

@XiongRui020325

data_process_amazon.py中train_target_item这个变量并没有被定义过,请问怎么修改?

Image

Activity

  1. hyp1231 commented on May 15, 2025

    @hyp1231
    Member

    感谢指出这个 bug!我们刚刚进行了修正 5ec1154

    只要把这行删掉即可。

  2. XiongRui020325 commented on May 16, 2025

    @XiongRui020325
    Author

    好的,想问下我目前正在尝试把这个模型应用到Amazon Review 2023的数据集上,目前数据预处理阶段python data_process_amazon.py --dataset Games已经完成了,在运行python sample_candidates.py -d Games -u 6000 -s random就报错了:
    Image
    想问下是什么原因?以及Games-6k是从何而来的?

  3. hyp1231 commented on May 16, 2025

    @hyp1231
    Member

    你好,采样数据这里需要加载预训练模型 checkpoint,我们提供的预训练 checkpoints 是在 2018 数据集上预训练的。

    所以如果想应用到 Amazon Reviews 2023 数据集上,需要重新预训练几个基础模型,比如 SASRec, GRU4Rec, BPR, Pop。可以参考 https://github.com/RUCAIBox/LLMRank/blob/master/scripts/ob1-struggle-to-perceive-order-but-can-be-triggered.md#conventional-methods 下面被注释掉的那些指令,比如如果希望预训练 SASRec,则:

    python run_baseline.py -m SASRec -d Games
    mv xxx.pth pretrained_models/SASRec-Games.pth # 请把 xxx.pth 替换成对应 checkpoint 路径

    这里的 Games 应该是 2023 版本的 Games 数据集(不过因为我也没试过 2023 数据集,可能也会有适配问题需要解决)。

    关于为什么是 6k,是因为 Games 原始数据有点大,所以我们采样了 6000 个用户使数据集的大小和 ml-1m 差不多。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions