nextrec 0.5.1__tar.gz → 0.5.3__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (194) hide show
  1. {nextrec-0.5.1 → nextrec-0.5.3}/PKG-INFO +4 -5
  2. {nextrec-0.5.1 → nextrec-0.5.3}/README.md +3 -4
  3. {nextrec-0.5.1 → nextrec-0.5.3}/README_en.md +3 -4
  4. {nextrec-0.5.1 → nextrec-0.5.3}/docs/rtd/conf.py +1 -1
  5. nextrec-0.5.3/nextrec/__version__.py +1 -0
  6. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/basic/model.py +288 -181
  7. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/basic/summary.py +21 -4
  8. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/cli.py +35 -15
  9. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/data/batch_utils.py +1 -1
  10. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/data/data_processing.py +1 -35
  11. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/data/data_utils.py +0 -4
  12. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/data/dataloader.py +125 -103
  13. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/data/preprocessor.py +141 -92
  14. nextrec-0.5.3/nextrec/models/multi_task/__init__.py +0 -0
  15. nextrec-0.5.3/nextrec/models/ranking/__init__.py +0 -0
  16. nextrec-0.5.3/nextrec/models/retrieval/__init__.py +0 -0
  17. nextrec-0.5.3/nextrec/models/tree_base/__init__.py +0 -0
  18. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/tree_base/base.py +14 -23
  19. nextrec-0.5.3/nextrec/utils/__init__.py +0 -0
  20. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/utils/data.py +39 -119
  21. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/utils/model.py +5 -14
  22. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/NextRec-CLI.md +8 -5
  23. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/NextRec-CLI_zh.md +8 -5
  24. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/predict_config.yaml +5 -2
  25. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/predict_config_template.yaml +5 -2
  26. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/train_config_template.yaml +2 -2
  27. {nextrec-0.5.1 → nextrec-0.5.3}/pyproject.toml +1 -1
  28. nextrec-0.5.3/scripts/profile_predict.py +521 -0
  29. {nextrec-0.5.1 → nextrec-0.5.3}/test/test_losses.py +3 -6
  30. {nextrec-0.5.1 → nextrec-0.5.3}/test/test_preprocessor.py +0 -31
  31. {nextrec-0.5.1 → nextrec-0.5.3}/test/test_utils_data.py +5 -52
  32. {nextrec-0.5.1 → nextrec-0.5.3}/tutorials/distributed/example_distributed_training.py +1 -1
  33. {nextrec-0.5.1 → nextrec-0.5.3}/tutorials/example_match.py +2 -2
  34. {nextrec-0.5.1 → nextrec-0.5.3}/tutorials/example_multitask.py +1 -1
  35. {nextrec-0.5.1 → nextrec-0.5.3}/tutorials/movielen_match_dssm.py +2 -3
  36. {nextrec-0.5.1 → nextrec-0.5.3}/tutorials/run_all_match_models.py +2 -1
  37. {nextrec-0.5.1 → nextrec-0.5.3}/tutorials/run_all_multitask_models.py +1 -1
  38. {nextrec-0.5.1 → nextrec-0.5.3}/tutorials/run_all_ranking_models.py +1 -1
  39. nextrec-0.5.1/nextrec/__version__.py +0 -1
  40. nextrec-0.5.1/nextrec/data/__init__.py +0 -52
  41. nextrec-0.5.1/nextrec/loss/__init__.py +0 -36
  42. nextrec-0.5.1/nextrec/models/generative/__init__.py +0 -9
  43. nextrec-0.5.1/nextrec/models/tree_base/__init__.py +0 -15
  44. nextrec-0.5.1/nextrec/utils/__init__.py +0 -119
  45. {nextrec-0.5.1 → nextrec-0.5.3}/.github/workflows/publish.yml +0 -0
  46. {nextrec-0.5.1 → nextrec-0.5.3}/.github/workflows/tests.yml +0 -0
  47. {nextrec-0.5.1 → nextrec-0.5.3}/.gitignore +0 -0
  48. {nextrec-0.5.1 → nextrec-0.5.3}/.readthedocs.yaml +0 -0
  49. {nextrec-0.5.1 → nextrec-0.5.3}/CODE_OF_CONDUCT.md +0 -0
  50. {nextrec-0.5.1 → nextrec-0.5.3}/CONTRIBUTING.md +0 -0
  51. {nextrec-0.5.1 → nextrec-0.5.3}/LICENSE +0 -0
  52. {nextrec-0.5.1 → nextrec-0.5.3}/MANIFEST.in +0 -0
  53. {nextrec-0.5.1 → nextrec-0.5.3}/assets/Feature Configuration.png +0 -0
  54. {nextrec-0.5.1 → nextrec-0.5.3}/assets/Model Parameters.png +0 -0
  55. {nextrec-0.5.1 → nextrec-0.5.3}/assets/Training Configuration.png +0 -0
  56. {nextrec-0.5.1 → nextrec-0.5.3}/assets/Training logs.png +0 -0
  57. {nextrec-0.5.1 → nextrec-0.5.3}/assets/logo.png +0 -0
  58. {nextrec-0.5.1 → nextrec-0.5.3}/assets/mmoe_tutorial.png +0 -0
  59. {nextrec-0.5.1 → nextrec-0.5.3}/assets/nextrec_diagram.png +0 -0
  60. {nextrec-0.5.1 → nextrec-0.5.3}/assets/test data.png +0 -0
  61. {nextrec-0.5.1 → nextrec-0.5.3}/dataset/ctcvr_task.csv +0 -0
  62. {nextrec-0.5.1 → nextrec-0.5.3}/dataset/ecommerce_task.csv +0 -0
  63. {nextrec-0.5.1 → nextrec-0.5.3}/dataset/match_task.csv +0 -0
  64. {nextrec-0.5.1 → nextrec-0.5.3}/dataset/movielens_100k.csv +0 -0
  65. {nextrec-0.5.1 → nextrec-0.5.3}/dataset/multitask_task.csv +0 -0
  66. {nextrec-0.5.1 → nextrec-0.5.3}/dataset/ranking_task.csv +0 -0
  67. {nextrec-0.5.1 → nextrec-0.5.3}/docs/en/Getting started guide.md +0 -0
  68. {nextrec-0.5.1 → nextrec-0.5.3}/docs/rtd/Makefile +0 -0
  69. {nextrec-0.5.1 → nextrec-0.5.3}/docs/rtd/index.md +0 -0
  70. {nextrec-0.5.1 → nextrec-0.5.3}/docs/rtd/make.bat +0 -0
  71. {nextrec-0.5.1 → nextrec-0.5.3}/docs/rtd/modules.rst +0 -0
  72. {nextrec-0.5.1 → nextrec-0.5.3}/docs/rtd/nextrec.basic.rst +0 -0
  73. {nextrec-0.5.1 → nextrec-0.5.3}/docs/rtd/nextrec.data.rst +0 -0
  74. {nextrec-0.5.1 → nextrec-0.5.3}/docs/rtd/nextrec.loss.rst +0 -0
  75. {nextrec-0.5.1 → nextrec-0.5.3}/docs/rtd/nextrec.rst +0 -0
  76. {nextrec-0.5.1 → nextrec-0.5.3}/docs/rtd/nextrec.utils.rst +0 -0
  77. {nextrec-0.5.1 → nextrec-0.5.3}/docs/rtd/requirements.txt +0 -0
  78. {nextrec-0.5.1 → nextrec-0.5.3}/docs/zh//345/277/253/351/200/237/344/270/212/346/211/213.md" +0 -0
  79. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/__init__.py +0 -0
  80. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/basic/__init__.py +0 -0
  81. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/basic/activation.py +0 -0
  82. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/basic/asserts.py +0 -0
  83. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/basic/callback.py +0 -0
  84. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/basic/features.py +0 -0
  85. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/basic/heads.py +0 -0
  86. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/basic/layers.py +0 -0
  87. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/basic/loggers.py +0 -0
  88. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/basic/metrics.py +0 -0
  89. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/basic/session.py +0 -0
  90. {nextrec-0.5.1/nextrec/models/multi_task → nextrec-0.5.3/nextrec/data}/__init__.py +0 -0
  91. {nextrec-0.5.1/nextrec/models/ranking → nextrec-0.5.3/nextrec/loss}/__init__.py +0 -0
  92. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/loss/grad_norm.py +0 -0
  93. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/loss/listwise.py +0 -0
  94. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/loss/pairwise.py +0 -0
  95. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/loss/pointwise.py +0 -0
  96. {nextrec-0.5.1/nextrec/models/retrieval → nextrec-0.5.3/nextrec/models/generative}/__init__.py +0 -0
  97. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/generative/tiger.py +0 -0
  98. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/multi_task/[pre]aitm.py +0 -0
  99. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/multi_task/[pre]snr_trans.py +0 -0
  100. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/multi_task/apg.py +0 -0
  101. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/multi_task/cross_stitch.py +0 -0
  102. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/multi_task/escm.py +0 -0
  103. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/multi_task/esmm.py +0 -0
  104. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/multi_task/hmoe.py +0 -0
  105. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/multi_task/mmoe.py +0 -0
  106. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/multi_task/pepnet.py +0 -0
  107. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/multi_task/ple.py +0 -0
  108. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/multi_task/poso.py +0 -0
  109. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/multi_task/share_bottom.py +0 -0
  110. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/ranking/afm.py +0 -0
  111. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/ranking/autoint.py +0 -0
  112. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/ranking/dcn.py +0 -0
  113. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/ranking/dcn_v2.py +0 -0
  114. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/ranking/deepfm.py +0 -0
  115. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/ranking/dien.py +0 -0
  116. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/ranking/din.py +0 -0
  117. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/ranking/eulernet.py +0 -0
  118. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/ranking/ffm.py +0 -0
  119. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/ranking/fibinet.py +0 -0
  120. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/ranking/fm.py +0 -0
  121. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/ranking/lr.py +0 -0
  122. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/ranking/masknet.py +0 -0
  123. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/ranking/pnn.py +0 -0
  124. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/ranking/widedeep.py +0 -0
  125. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/ranking/xdeepfm.py +0 -0
  126. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/representation/__init__.py +0 -0
  127. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/representation/rqvae.py +0 -0
  128. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/retrieval/dssm.py +0 -0
  129. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/retrieval/dssm_v2.py +0 -0
  130. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/retrieval/mind.py +0 -0
  131. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/retrieval/sdm.py +0 -0
  132. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/retrieval/youtube_dnn.py +0 -0
  133. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/sequential/hstu.py +0 -0
  134. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/tree_base/catboost.py +0 -0
  135. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/tree_base/lightgbm.py +0 -0
  136. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/models/tree_base/xgboost.py +0 -0
  137. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/utils/config.py +0 -0
  138. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/utils/console.py +0 -0
  139. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/utils/embedding.py +0 -0
  140. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/utils/loss.py +0 -0
  141. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/utils/onnx_utils.py +0 -0
  142. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/utils/torch_utils.py +0 -0
  143. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec/utils/types.py +0 -0
  144. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/feature_config.yaml +0 -0
  145. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/afm.yaml +0 -0
  146. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/apg.yaml +0 -0
  147. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/autoint.yaml +0 -0
  148. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/cross_stitch.yaml +0 -0
  149. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/dcn.yaml +0 -0
  150. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/deepfm.yaml +0 -0
  151. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/din.yaml +0 -0
  152. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/escm.yaml +0 -0
  153. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/esmm.yaml +0 -0
  154. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/fibinet.yaml +0 -0
  155. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/fm.yaml +0 -0
  156. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/hmoe.yaml +0 -0
  157. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/masknet.yaml +0 -0
  158. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/mmoe.yaml +0 -0
  159. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/pepnet.yaml +0 -0
  160. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/ple.yaml +0 -0
  161. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/pnn.yaml +0 -0
  162. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/poso.yaml +0 -0
  163. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/share_bottom.yaml +0 -0
  164. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/widedeep.yaml +0 -0
  165. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/xdeepfm.yaml +0 -0
  166. {nextrec-0.5.1 → nextrec-0.5.3}/nextrec_cli_preset/train_config.yaml +0 -0
  167. {nextrec-0.5.1 → nextrec-0.5.3}/pytest.ini +0 -0
  168. {nextrec-0.5.1 → nextrec-0.5.3}/requirements.txt +0 -0
  169. {nextrec-0.5.1 → nextrec-0.5.3}/scripts/format_code.py +0 -0
  170. {nextrec-0.5.1 → nextrec-0.5.3}/test/__init__.py +0 -0
  171. {nextrec-0.5.1 → nextrec-0.5.3}/test/conftest.py +0 -0
  172. {nextrec-0.5.1 → nextrec-0.5.3}/test/helpers.py +0 -0
  173. {nextrec-0.5.1 → nextrec-0.5.3}/test/run_tests.py +0 -0
  174. {nextrec-0.5.1 → nextrec-0.5.3}/test/test_base_model_regularization.py +0 -0
  175. {nextrec-0.5.1 → nextrec-0.5.3}/test/test_generative_models.py +0 -0
  176. {nextrec-0.5.1 → nextrec-0.5.3}/test/test_layers.py +0 -0
  177. {nextrec-0.5.1 → nextrec-0.5.3}/test/test_match_models.py +0 -0
  178. {nextrec-0.5.1 → nextrec-0.5.3}/test/test_multitask_models.py +0 -0
  179. {nextrec-0.5.1 → nextrec-0.5.3}/test/test_onnx_models.py +0 -0
  180. {nextrec-0.5.1 → nextrec-0.5.3}/test/test_ranking_models.py +0 -0
  181. {nextrec-0.5.1 → nextrec-0.5.3}/test/test_utils_console.py +0 -0
  182. {nextrec-0.5.1 → nextrec-0.5.3}/test/test_utils_embedding.py +0 -0
  183. {nextrec-0.5.1 → nextrec-0.5.3}/test_requirements.txt +0 -0
  184. {nextrec-0.5.1 → nextrec-0.5.3}/tutorials/distributed/example_distributed_training_large_dataset.py +0 -0
  185. {nextrec-0.5.1 → nextrec-0.5.3}/tutorials/example_onnx.py +0 -0
  186. {nextrec-0.5.1 → nextrec-0.5.3}/tutorials/example_ranking_din.py +0 -0
  187. {nextrec-0.5.1 → nextrec-0.5.3}/tutorials/example_tree.py +0 -0
  188. {nextrec-0.5.1 → nextrec-0.5.3}/tutorials/movielen_ranking_deepfm.py +0 -0
  189. {nextrec-0.5.1 → nextrec-0.5.3}/tutorials/notebooks/en/Build semantic ID with RQ-VAE.ipynb +0 -0
  190. {nextrec-0.5.1 → nextrec-0.5.3}/tutorials/notebooks/en/Hands on dataprocessor.ipynb +0 -0
  191. {nextrec-0.5.1 → nextrec-0.5.3}/tutorials/notebooks/en/Hands on nextrec.ipynb +0 -0
  192. {nextrec-0.5.1 → nextrec-0.5.3}/tutorials/notebooks/zh//344/275/277/347/224/250RQ-VAE/346/236/204/345/273/272/350/257/255/344/271/211ID.ipynb" +0 -0
  193. {nextrec-0.5.1 → nextrec-0.5.3}/tutorials/notebooks/zh//345/246/202/344/275/225/344/275/277/347/224/250DataProcessor/350/277/233/350/241/214/351/242/204/345/244/204/347/220/206.ipynb" +0 -0
  194. {nextrec-0.5.1 → nextrec-0.5.3}/tutorials/notebooks/zh//345/277/253/351/200/237/345/205/245/351/227/250nextrec.ipynb" +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: nextrec
3
- Version: 0.5.1
3
+ Version: 0.5.3
4
4
  Summary: A comprehensive recommendation library with match, ranking, and multi-task learning models
5
5
  Project-URL: Homepage, https://github.com/zerolovesea/NextRec
6
6
  Project-URL: Repository, https://github.com/zerolovesea/NextRec
@@ -73,7 +73,7 @@ Description-Content-Type: text/markdown
73
73
  ![Python](https://img.shields.io/badge/Python-3.10+-blue.svg)
74
74
  ![PyTorch](https://img.shields.io/badge/PyTorch-1.10+-ee4c2c.svg)
75
75
  ![License](https://img.shields.io/badge/License-Apache%202.0-green.svg)
76
- ![Version](https://img.shields.io/badge/Version-0.5.1-orange.svg)
76
+ ![Version](https://img.shields.io/badge/Version-0.5.3-orange.svg)
77
77
  [![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
78
78
 
79
79
  中文文档 | [English Version](README_en.md)
@@ -112,7 +112,6 @@ NextRec是一个基于PyTorch的现代推荐系统框架,旨在为研究工程
112
112
  - **21/12/2025** 在v0.4.16中加入了对[GradNorm](/nextrec/loss/grad_norm.py)的支持,通过compile的`loss_weight='grad_norm'`进行配置
113
113
  - **12/12/2025** 在v0.4.9中加入了[RQ-VAE](/nextrec/models/representation/rqvae.py)模块。配套的[数据集](/dataset/ecommerce_task.csv)和[代码](tutorials/notebooks/zh/使用RQ-VAE构建语义ID.ipynb)已经同步在仓库中
114
114
  - **07/12/2025** 发布了NextRec CLI命令行工具,它允许用户根据配置文件进行一键训练和推理,我们提供了相关的[教程](/nextrec_cli_preset/NextRec-CLI_zh.md)和[教学代码](/nextrec_cli_preset)
115
- - **03/12/2025** NextRec获得了100颗🌟!感谢大家的支持
116
115
  - **06/12/2025** 在v0.4.1中支持了单机多卡的分布式DDP训练,并且提供了配套的[代码](tutorials/distributed)
117
116
  - **11/11/2025** NextRec v0.1.0发布,我们提供了10余种Ranking模型,11种多任务模型和4种召回模型,以及统一的训练/日志/指标管理系统
118
117
 
@@ -260,11 +259,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
260
259
 
261
260
  预测结果固定保存到 `{checkpoint_path}/predictions/{name}.{save_data_format}`。
262
261
 
263
- > 截止当前版本0.5.1,NextRec CLI支持单机训练,分布式训练相关功能尚在开发中。
262
+ > 截止当前版本0.5.3,NextRec CLI支持单机训练,分布式训练相关功能尚在开发中。
264
263
 
265
264
  ## 兼容平台
266
265
 
267
- 当前最新版本为0.5.1,所有模型和测试代码均已在以下平台通过验证,如果开发者在使用中遇到兼容问题,请在issue区提出错误报告及系统版本:
266
+ 当前最新版本为0.5.3,所有模型和测试代码均已在以下平台通过验证,如果开发者在使用中遇到兼容问题,请在issue区提出错误报告及系统版本:
268
267
 
269
268
  | 平台 | 配置 |
270
269
  |------|------|
@@ -8,7 +8,7 @@
8
8
  ![Python](https://img.shields.io/badge/Python-3.10+-blue.svg)
9
9
  ![PyTorch](https://img.shields.io/badge/PyTorch-1.10+-ee4c2c.svg)
10
10
  ![License](https://img.shields.io/badge/License-Apache%202.0-green.svg)
11
- ![Version](https://img.shields.io/badge/Version-0.5.1-orange.svg)
11
+ ![Version](https://img.shields.io/badge/Version-0.5.3-orange.svg)
12
12
  [![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
13
13
 
14
14
  中文文档 | [English Version](README_en.md)
@@ -47,7 +47,6 @@ NextRec是一个基于PyTorch的现代推荐系统框架,旨在为研究工程
47
47
  - **21/12/2025** 在v0.4.16中加入了对[GradNorm](/nextrec/loss/grad_norm.py)的支持,通过compile的`loss_weight='grad_norm'`进行配置
48
48
  - **12/12/2025** 在v0.4.9中加入了[RQ-VAE](/nextrec/models/representation/rqvae.py)模块。配套的[数据集](/dataset/ecommerce_task.csv)和[代码](tutorials/notebooks/zh/使用RQ-VAE构建语义ID.ipynb)已经同步在仓库中
49
49
  - **07/12/2025** 发布了NextRec CLI命令行工具,它允许用户根据配置文件进行一键训练和推理,我们提供了相关的[教程](/nextrec_cli_preset/NextRec-CLI_zh.md)和[教学代码](/nextrec_cli_preset)
50
- - **03/12/2025** NextRec获得了100颗🌟!感谢大家的支持
51
50
  - **06/12/2025** 在v0.4.1中支持了单机多卡的分布式DDP训练,并且提供了配套的[代码](tutorials/distributed)
52
51
  - **11/11/2025** NextRec v0.1.0发布,我们提供了10余种Ranking模型,11种多任务模型和4种召回模型,以及统一的训练/日志/指标管理系统
53
52
 
@@ -195,11 +194,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
195
194
 
196
195
  预测结果固定保存到 `{checkpoint_path}/predictions/{name}.{save_data_format}`。
197
196
 
198
- > 截止当前版本0.5.1,NextRec CLI支持单机训练,分布式训练相关功能尚在开发中。
197
+ > 截止当前版本0.5.3,NextRec CLI支持单机训练,分布式训练相关功能尚在开发中。
199
198
 
200
199
  ## 兼容平台
201
200
 
202
- 当前最新版本为0.5.1,所有模型和测试代码均已在以下平台通过验证,如果开发者在使用中遇到兼容问题,请在issue区提出错误报告及系统版本:
201
+ 当前最新版本为0.5.3,所有模型和测试代码均已在以下平台通过验证,如果开发者在使用中遇到兼容问题,请在issue区提出错误报告及系统版本:
203
202
 
204
203
  | 平台 | 配置 |
205
204
  |------|------|
@@ -8,7 +8,7 @@
8
8
  ![Python](https://img.shields.io/badge/Python-3.10+-blue.svg)
9
9
  ![PyTorch](https://img.shields.io/badge/PyTorch-1.10+-ee4c2c.svg)
10
10
  ![License](https://img.shields.io/badge/License-Apache%202.0-green.svg)
11
- ![Version](https://img.shields.io/badge/Version-0.5.1-orange.svg)
11
+ ![Version](https://img.shields.io/badge/Version-0.5.3-orange.svg)
12
12
  [![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
13
13
 
14
14
  English | [中文文档](README.md)
@@ -50,7 +50,6 @@ NextRec is a modern recommendation framework built on PyTorch, delivering a unif
50
50
  - **21/12/2025** Added support for [GradNorm](/nextrec/loss/grad_norm.py) in v0.4.16, configurable via `loss_weight='grad_norm'` in the compile method
51
51
  - **12/12/2025** Added [RQ-VAE](/nextrec/models/representation/rqvae.py), a common module for generative retrieval in v0.4.9. Paired [dataset](/dataset/ecommerce_task.csv) and [notebook code](tutorials/notebooks/en/Build%20semantic%20ID%20with%20RQ-VAE.ipynb) are available.
52
52
  - **07/12/2025** Released the NextRec CLI tool to run training/inference from configs. See the [guide](/nextrec_cli_preset/NextRec-CLI.md) and [reference code](/nextrec_cli_preset).
53
- - **03/12/2025** NextRec reached 100 ⭐—thanks for the support!
54
53
  - **06/12/2025** Added single-machine multi-GPU DDP training in v0.4.1 with supporting [code](tutorials/distributed).
55
54
  - **11/11/2025** NextRec v0.1.0 released with 10+ ranking models, 11 multi-task models, 4 retrieval models, and a unified training/logging/metrics system.
56
55
 
@@ -198,11 +197,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
198
197
 
199
198
  Prediction outputs are saved under `{checkpoint_path}/predictions/{name}.{save_data_format}`.
200
199
 
201
- > As of version 0.5.1, NextRec CLI supports single-machine training; distributed training features are currently under development.
200
+ > As of version 0.5.3, NextRec CLI supports single-machine training; distributed training features are currently under development.
202
201
 
203
202
  ## Platform Compatibility
204
203
 
205
- The current version is 0.5.1. All models and test code have been validated on the following platforms. If you encounter compatibility issues, please report them in the issue tracker with your system version:
204
+ The current version is 0.5.3. All models and test code have been validated on the following platforms. If you encounter compatibility issues, please report them in the issue tracker with your system version:
206
205
 
207
206
  | Platform | Configuration |
208
207
  |----------|---------------|
@@ -11,7 +11,7 @@ sys.path.insert(0, str(PROJECT_ROOT / "nextrec"))
11
11
  project = "NextRec"
12
12
  copyright = "2026, Yang Zhou"
13
13
  author = "Yang Zhou"
14
- release = "0.5.1"
14
+ release = "0.5.3"
15
15
 
16
16
  extensions = [
17
17
  "myst_parser",
@@ -0,0 +1 @@
1
+ __version__ = "0.5.3"