NVIDIA-Merlin / HugeCTR

HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training
Apache License 2.0
946 stars 200 forks source link

[Question] negative item counts in MovieLens notebook #384

Closed JohnFirth closed 1 year ago

JohnFirth commented 1 year ago

Hey, can you explain the choices of 500 and 100 negative items per user for the train and test sets here https://github.com/NVIDIA-Merlin/HugeCTR/blob/main/notebooks/movie-lens-example.ipynb ?

Thanks!

jershi425 commented 1 year ago

Hi @JohnFirth , thanks for the question. The negative items per user for train and test sets are chosen arbitrarily.