NVIDIA-Merlin / Merlin

NVIDIA Merlin is an open source library providing end-to-end GPU-accelerated recommender systems, from feature engineering and preprocessing to training deep learning models and running inference in production.
Apache License 2.0
722 stars 112 forks source link

Improvements in Quick-start for Ranking #1014

Closed gabrielspmoreira closed 1 year ago

gabrielspmoreira commented 1 year ago

This PR adds some improvements to the Quick-start for ranking scripts and documentation

In preprocessing.py:

In ranking.py:

This PR also converts those scripts to Python modules, to make it easier to import/extend their classes and to test them. So now, instead of being run like python preprocessing.py --args ..., they need to be run as a Python module, e.g.

cd /Merlin/examples/
python -m quick_start.scripts.preproc.preprocessing --args ...
github-actions[bot] commented 1 year ago

Documentation preview

https://nvidia-merlin.github.io/Merlin/review/pr-1014

rnyak commented 1 year ago

@gabrielspmoreira I approved in case you want to merge once you push your final changes.