Skip to content

SASRec perform better in the ml-1m provided in this project #9

@kkkkobe7

Description

@kkkkobe7

I want to know if the ml-1m dataset in this project and the ml-1m dataset in SASRec.pytorch are from the same source?

I have tested both projects separately, it seems right. But when I directly removed the 'rate' and 'timestamp' from each row of the ml-1m.txt in this project and tested SASRec, the results were as follows

NCDG@10: 0.8290 HR@10: 0.9831

The results are far from what was reported in the paper.

Both codes used the hyperparameters from the paper, specifically: epoch 200, batch 128, regularization 0.0005, maxlen 50, max_time intervals 2048

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions