Skip to content
Permalink

Comparing changes

Choose two branches to see what’s changed or to start a new pull request. If you need to, you can also or learn more about diff comparisons.

Open a pull request

Create a new pull request by comparing changes across two branches. If you need to, you can also . Learn more about diff comparisons here.
base repository: recommenders-team/recommenders
Failed to load repositories. Confirm that selected base ref is valid, then try again.
Loading
base: staging
Choose a base ref
...
head repository: recommenders-team/recommenders
Failed to load repositories. Confirm that selected head ref is valid, then try again.
Loading
compare: miguel/sasrec_unirec
Choose a head ref
Checking mergeability… Don’t worry, you can still create the pull request.
Loading
Showing with 7,177 additions and 19 deletions.
  1. +984 −0 examples/00_quick_start/sasrec_movielens.ipynb
  2. +65 −4 recommenders/datasets/pandas_df_utils.py
  3. +22 −0 recommenders/datasets/python_splitters.py
  4. +44 −1 recommenders/datasets/spark_splitters.py
  5. 0 recommenders/models/unirec/__init__.py
  6. +83 −0 recommenders/models/unirec/config/base.yaml
  7. +10 −0 recommenders/models/unirec/config/data/ml-100k.yaml
  8. +7 −0 recommenders/models/unirec/config/model/SASRec.yaml
  9. 0 recommenders/models/unirec/constants/__init__.py
  10. +9 −0 recommenders/models/unirec/constants/global_variables.py
  11. +18 −0 recommenders/models/unirec/constants/loss_funcs.py
  12. +88 −0 recommenders/models/unirec/constants/protocols.py
  13. 0 recommenders/models/unirec/data/__init__.py
  14. 0 recommenders/models/unirec/data/dataset/__init__.py
  15. +280 −0 recommenders/models/unirec/data/dataset/basedataset.py
  16. +56 −0 recommenders/models/unirec/data/dataset/movielens_utils.py
  17. +70 −0 recommenders/models/unirec/data/dataset/seqrecdataset.py
  18. 0 recommenders/models/unirec/facility/__init__.py
  19. 0 recommenders/models/unirec/facility/evaluation/__init__.py
  20. +301 −0 recommenders/models/unirec/facility/evaluation/evaluator_abc.py
  21. +281 −0 recommenders/models/unirec/facility/evaluation/multipos.py
  22. +258 −0 recommenders/models/unirec/facility/evaluation/onepos.py
  23. +273 −0 recommenders/models/unirec/facility/evaluation/sessionwise.py
  24. +124 −0 recommenders/models/unirec/facility/morec/__init__.py
  25. +214 −0 recommenders/models/unirec/facility/morec/_min_norm_solver.py
  26. +472 −0 recommenders/models/unirec/facility/morec/morec_data_sampler.py
  27. +405 −0 recommenders/models/unirec/facility/morec/morec_objective_controller.py
  28. +53 −0 recommenders/models/unirec/facility/solver.py
  29. +598 −0 recommenders/models/unirec/facility/trainer.py
  30. 0 recommenders/models/unirec/model/__init__.py
  31. 0 recommenders/models/unirec/model/base/__init__.py
  32. +353 −0 recommenders/models/unirec/model/base/reco_abc.py
  33. +273 −0 recommenders/models/unirec/model/base/recommender.py
  34. +885 −0 recommenders/models/unirec/model/modules.py
  35. 0 recommenders/models/unirec/model/sequential/__init__.py
  36. +103 −0 recommenders/models/unirec/model/sequential/sasrec.py
  37. +28 −0 recommenders/models/unirec/model/sequential/seqrec_base.py
  38. 0 recommenders/models/unirec/utils/__init__.py
  39. +175 −0 recommenders/models/unirec/utils/file_io.py
  40. +288 −0 recommenders/models/unirec/utils/general.py
  41. +2 −0 setup.py
  42. +13 −1 tests/conftest.py
  43. +34 −0 tests/unit/recommenders/datasets/test_pandas_df_utils.py
  44. +65 −10 tests/unit/recommenders/datasets/test_python_splitter.py
  45. +95 −3 tests/unit/recommenders/datasets/test_spark_splitter.py
  46. +148 −0 tests/unit/recommenders/models/test_unirec_model.py
Loading