nextrec 0.5.2__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 (189) hide show
  1. {nextrec-0.5.2 → nextrec-0.5.3}/PKG-INFO +4 -4
  2. {nextrec-0.5.2 → nextrec-0.5.3}/README.md +3 -3
  3. {nextrec-0.5.2 → nextrec-0.5.3}/README_en.md +3 -3
  4. {nextrec-0.5.2 → nextrec-0.5.3}/docs/rtd/conf.py +1 -1
  5. nextrec-0.5.3/nextrec/__version__.py +1 -0
  6. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/basic/model.py +1 -1
  7. {nextrec-0.5.2 → nextrec-0.5.3}/pyproject.toml +1 -1
  8. nextrec-0.5.2/nextrec/__version__.py +0 -1
  9. {nextrec-0.5.2 → nextrec-0.5.3}/.github/workflows/publish.yml +0 -0
  10. {nextrec-0.5.2 → nextrec-0.5.3}/.github/workflows/tests.yml +0 -0
  11. {nextrec-0.5.2 → nextrec-0.5.3}/.gitignore +0 -0
  12. {nextrec-0.5.2 → nextrec-0.5.3}/.readthedocs.yaml +0 -0
  13. {nextrec-0.5.2 → nextrec-0.5.3}/CODE_OF_CONDUCT.md +0 -0
  14. {nextrec-0.5.2 → nextrec-0.5.3}/CONTRIBUTING.md +0 -0
  15. {nextrec-0.5.2 → nextrec-0.5.3}/LICENSE +0 -0
  16. {nextrec-0.5.2 → nextrec-0.5.3}/MANIFEST.in +0 -0
  17. {nextrec-0.5.2 → nextrec-0.5.3}/assets/Feature Configuration.png +0 -0
  18. {nextrec-0.5.2 → nextrec-0.5.3}/assets/Model Parameters.png +0 -0
  19. {nextrec-0.5.2 → nextrec-0.5.3}/assets/Training Configuration.png +0 -0
  20. {nextrec-0.5.2 → nextrec-0.5.3}/assets/Training logs.png +0 -0
  21. {nextrec-0.5.2 → nextrec-0.5.3}/assets/logo.png +0 -0
  22. {nextrec-0.5.2 → nextrec-0.5.3}/assets/mmoe_tutorial.png +0 -0
  23. {nextrec-0.5.2 → nextrec-0.5.3}/assets/nextrec_diagram.png +0 -0
  24. {nextrec-0.5.2 → nextrec-0.5.3}/assets/test data.png +0 -0
  25. {nextrec-0.5.2 → nextrec-0.5.3}/dataset/ctcvr_task.csv +0 -0
  26. {nextrec-0.5.2 → nextrec-0.5.3}/dataset/ecommerce_task.csv +0 -0
  27. {nextrec-0.5.2 → nextrec-0.5.3}/dataset/match_task.csv +0 -0
  28. {nextrec-0.5.2 → nextrec-0.5.3}/dataset/movielens_100k.csv +0 -0
  29. {nextrec-0.5.2 → nextrec-0.5.3}/dataset/multitask_task.csv +0 -0
  30. {nextrec-0.5.2 → nextrec-0.5.3}/dataset/ranking_task.csv +0 -0
  31. {nextrec-0.5.2 → nextrec-0.5.3}/docs/en/Getting started guide.md +0 -0
  32. {nextrec-0.5.2 → nextrec-0.5.3}/docs/rtd/Makefile +0 -0
  33. {nextrec-0.5.2 → nextrec-0.5.3}/docs/rtd/index.md +0 -0
  34. {nextrec-0.5.2 → nextrec-0.5.3}/docs/rtd/make.bat +0 -0
  35. {nextrec-0.5.2 → nextrec-0.5.3}/docs/rtd/modules.rst +0 -0
  36. {nextrec-0.5.2 → nextrec-0.5.3}/docs/rtd/nextrec.basic.rst +0 -0
  37. {nextrec-0.5.2 → nextrec-0.5.3}/docs/rtd/nextrec.data.rst +0 -0
  38. {nextrec-0.5.2 → nextrec-0.5.3}/docs/rtd/nextrec.loss.rst +0 -0
  39. {nextrec-0.5.2 → nextrec-0.5.3}/docs/rtd/nextrec.rst +0 -0
  40. {nextrec-0.5.2 → nextrec-0.5.3}/docs/rtd/nextrec.utils.rst +0 -0
  41. {nextrec-0.5.2 → nextrec-0.5.3}/docs/rtd/requirements.txt +0 -0
  42. {nextrec-0.5.2 → nextrec-0.5.3}/docs/zh//345/277/253/351/200/237/344/270/212/346/211/213.md" +0 -0
  43. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/__init__.py +0 -0
  44. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/basic/__init__.py +0 -0
  45. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/basic/activation.py +0 -0
  46. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/basic/asserts.py +0 -0
  47. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/basic/callback.py +0 -0
  48. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/basic/features.py +0 -0
  49. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/basic/heads.py +0 -0
  50. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/basic/layers.py +0 -0
  51. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/basic/loggers.py +0 -0
  52. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/basic/metrics.py +0 -0
  53. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/basic/session.py +0 -0
  54. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/basic/summary.py +0 -0
  55. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/cli.py +0 -0
  56. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/data/__init__.py +0 -0
  57. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/data/batch_utils.py +0 -0
  58. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/data/data_processing.py +0 -0
  59. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/data/data_utils.py +0 -0
  60. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/data/dataloader.py +0 -0
  61. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/data/preprocessor.py +0 -0
  62. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/loss/__init__.py +0 -0
  63. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/loss/grad_norm.py +0 -0
  64. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/loss/listwise.py +0 -0
  65. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/loss/pairwise.py +0 -0
  66. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/loss/pointwise.py +0 -0
  67. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/generative/__init__.py +0 -0
  68. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/generative/tiger.py +0 -0
  69. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/multi_task/[pre]aitm.py +0 -0
  70. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/multi_task/[pre]snr_trans.py +0 -0
  71. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/multi_task/__init__.py +0 -0
  72. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/multi_task/apg.py +0 -0
  73. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/multi_task/cross_stitch.py +0 -0
  74. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/multi_task/escm.py +0 -0
  75. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/multi_task/esmm.py +0 -0
  76. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/multi_task/hmoe.py +0 -0
  77. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/multi_task/mmoe.py +0 -0
  78. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/multi_task/pepnet.py +0 -0
  79. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/multi_task/ple.py +0 -0
  80. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/multi_task/poso.py +0 -0
  81. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/multi_task/share_bottom.py +0 -0
  82. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/ranking/__init__.py +0 -0
  83. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/ranking/afm.py +0 -0
  84. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/ranking/autoint.py +0 -0
  85. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/ranking/dcn.py +0 -0
  86. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/ranking/dcn_v2.py +0 -0
  87. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/ranking/deepfm.py +0 -0
  88. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/ranking/dien.py +0 -0
  89. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/ranking/din.py +0 -0
  90. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/ranking/eulernet.py +0 -0
  91. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/ranking/ffm.py +0 -0
  92. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/ranking/fibinet.py +0 -0
  93. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/ranking/fm.py +0 -0
  94. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/ranking/lr.py +0 -0
  95. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/ranking/masknet.py +0 -0
  96. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/ranking/pnn.py +0 -0
  97. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/ranking/widedeep.py +0 -0
  98. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/ranking/xdeepfm.py +0 -0
  99. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/representation/__init__.py +0 -0
  100. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/representation/rqvae.py +0 -0
  101. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/retrieval/__init__.py +0 -0
  102. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/retrieval/dssm.py +0 -0
  103. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/retrieval/dssm_v2.py +0 -0
  104. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/retrieval/mind.py +0 -0
  105. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/retrieval/sdm.py +0 -0
  106. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/retrieval/youtube_dnn.py +0 -0
  107. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/sequential/hstu.py +0 -0
  108. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/tree_base/__init__.py +0 -0
  109. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/tree_base/base.py +0 -0
  110. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/tree_base/catboost.py +0 -0
  111. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/tree_base/lightgbm.py +0 -0
  112. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/models/tree_base/xgboost.py +0 -0
  113. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/utils/__init__.py +0 -0
  114. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/utils/config.py +0 -0
  115. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/utils/console.py +0 -0
  116. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/utils/data.py +0 -0
  117. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/utils/embedding.py +0 -0
  118. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/utils/loss.py +0 -0
  119. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/utils/model.py +0 -0
  120. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/utils/onnx_utils.py +0 -0
  121. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/utils/torch_utils.py +0 -0
  122. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec/utils/types.py +0 -0
  123. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/NextRec-CLI.md +0 -0
  124. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/NextRec-CLI_zh.md +0 -0
  125. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/feature_config.yaml +0 -0
  126. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/afm.yaml +0 -0
  127. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/apg.yaml +0 -0
  128. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/autoint.yaml +0 -0
  129. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/cross_stitch.yaml +0 -0
  130. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/dcn.yaml +0 -0
  131. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/deepfm.yaml +0 -0
  132. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/din.yaml +0 -0
  133. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/escm.yaml +0 -0
  134. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/esmm.yaml +0 -0
  135. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/fibinet.yaml +0 -0
  136. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/fm.yaml +0 -0
  137. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/hmoe.yaml +0 -0
  138. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/masknet.yaml +0 -0
  139. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/mmoe.yaml +0 -0
  140. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/pepnet.yaml +0 -0
  141. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/ple.yaml +0 -0
  142. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/pnn.yaml +0 -0
  143. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/poso.yaml +0 -0
  144. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/share_bottom.yaml +0 -0
  145. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/widedeep.yaml +0 -0
  146. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/model_configs/xdeepfm.yaml +0 -0
  147. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/predict_config.yaml +0 -0
  148. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/predict_config_template.yaml +0 -0
  149. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/train_config.yaml +0 -0
  150. {nextrec-0.5.2 → nextrec-0.5.3}/nextrec_cli_preset/train_config_template.yaml +0 -0
  151. {nextrec-0.5.2 → nextrec-0.5.3}/pytest.ini +0 -0
  152. {nextrec-0.5.2 → nextrec-0.5.3}/requirements.txt +0 -0
  153. {nextrec-0.5.2 → nextrec-0.5.3}/scripts/format_code.py +0 -0
  154. {nextrec-0.5.2 → nextrec-0.5.3}/scripts/profile_predict.py +0 -0
  155. {nextrec-0.5.2 → nextrec-0.5.3}/test/__init__.py +0 -0
  156. {nextrec-0.5.2 → nextrec-0.5.3}/test/conftest.py +0 -0
  157. {nextrec-0.5.2 → nextrec-0.5.3}/test/helpers.py +0 -0
  158. {nextrec-0.5.2 → nextrec-0.5.3}/test/run_tests.py +0 -0
  159. {nextrec-0.5.2 → nextrec-0.5.3}/test/test_base_model_regularization.py +0 -0
  160. {nextrec-0.5.2 → nextrec-0.5.3}/test/test_generative_models.py +0 -0
  161. {nextrec-0.5.2 → nextrec-0.5.3}/test/test_layers.py +0 -0
  162. {nextrec-0.5.2 → nextrec-0.5.3}/test/test_losses.py +0 -0
  163. {nextrec-0.5.2 → nextrec-0.5.3}/test/test_match_models.py +0 -0
  164. {nextrec-0.5.2 → nextrec-0.5.3}/test/test_multitask_models.py +0 -0
  165. {nextrec-0.5.2 → nextrec-0.5.3}/test/test_onnx_models.py +0 -0
  166. {nextrec-0.5.2 → nextrec-0.5.3}/test/test_preprocessor.py +0 -0
  167. {nextrec-0.5.2 → nextrec-0.5.3}/test/test_ranking_models.py +0 -0
  168. {nextrec-0.5.2 → nextrec-0.5.3}/test/test_utils_console.py +0 -0
  169. {nextrec-0.5.2 → nextrec-0.5.3}/test/test_utils_data.py +0 -0
  170. {nextrec-0.5.2 → nextrec-0.5.3}/test/test_utils_embedding.py +0 -0
  171. {nextrec-0.5.2 → nextrec-0.5.3}/test_requirements.txt +0 -0
  172. {nextrec-0.5.2 → nextrec-0.5.3}/tutorials/distributed/example_distributed_training.py +0 -0
  173. {nextrec-0.5.2 → nextrec-0.5.3}/tutorials/distributed/example_distributed_training_large_dataset.py +0 -0
  174. {nextrec-0.5.2 → nextrec-0.5.3}/tutorials/example_match.py +0 -0
  175. {nextrec-0.5.2 → nextrec-0.5.3}/tutorials/example_multitask.py +0 -0
  176. {nextrec-0.5.2 → nextrec-0.5.3}/tutorials/example_onnx.py +0 -0
  177. {nextrec-0.5.2 → nextrec-0.5.3}/tutorials/example_ranking_din.py +0 -0
  178. {nextrec-0.5.2 → nextrec-0.5.3}/tutorials/example_tree.py +0 -0
  179. {nextrec-0.5.2 → nextrec-0.5.3}/tutorials/movielen_match_dssm.py +0 -0
  180. {nextrec-0.5.2 → nextrec-0.5.3}/tutorials/movielen_ranking_deepfm.py +0 -0
  181. {nextrec-0.5.2 → nextrec-0.5.3}/tutorials/notebooks/en/Build semantic ID with RQ-VAE.ipynb +0 -0
  182. {nextrec-0.5.2 → nextrec-0.5.3}/tutorials/notebooks/en/Hands on dataprocessor.ipynb +0 -0
  183. {nextrec-0.5.2 → nextrec-0.5.3}/tutorials/notebooks/en/Hands on nextrec.ipynb +0 -0
  184. {nextrec-0.5.2 → 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
  185. {nextrec-0.5.2 → 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
  186. {nextrec-0.5.2 → nextrec-0.5.3}/tutorials/notebooks/zh//345/277/253/351/200/237/345/205/245/351/227/250nextrec.ipynb" +0 -0
  187. {nextrec-0.5.2 → nextrec-0.5.3}/tutorials/run_all_match_models.py +0 -0
  188. {nextrec-0.5.2 → nextrec-0.5.3}/tutorials/run_all_multitask_models.py +0 -0
  189. {nextrec-0.5.2 → nextrec-0.5.3}/tutorials/run_all_ranking_models.py +0 -0
{nextrec-0.5.2 → nextrec-0.5.3}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: nextrec
3
- Version: 0.5.2
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.2-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)
@@ -259,11 +259,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
259
259
 
260
260
  预测结果固定保存到 `{checkpoint_path}/predictions/{name}.{save_data_format}`。
261
261
 
262
- > 截止当前版本0.5.2，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
262
+ > 截止当前版本0.5.3，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
263
263
 
264
264
  ## 兼容平台
265
265
 
266
- 当前最新版本为0.5.2，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
266
+ 当前最新版本为0.5.3，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
267
267
 
268
268
  | 平台 | 配置 |
269
269
  |------|------|
{nextrec-0.5.2 → nextrec-0.5.3}/README.md RENAMED
@@ -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.2-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)
@@ -194,11 +194,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
194
194
 
195
195
  预测结果固定保存到 `{checkpoint_path}/predictions/{name}.{save_data_format}`。
196
196
 
197
- > 截止当前版本0.5.2，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
197
+ > 截止当前版本0.5.3，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
198
198
 
199
199
  ## 兼容平台
200
200
 
201
- 当前最新版本为0.5.2，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
201
+ 当前最新版本为0.5.3，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
202
202
 
203
203
  | 平台 | 配置 |
204
204
  |------|------|
{nextrec-0.5.2 → nextrec-0.5.3}/README_en.md RENAMED
@@ -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.2-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)
@@ -197,11 +197,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
197
197
 
198
198
  Prediction outputs are saved under `{checkpoint_path}/predictions/{name}.{save_data_format}`.
199
199
 
200
- > As of version 0.5.2, 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.
201
201
 
202
202
  ## Platform Compatibility
203
203
 
204
- The current version is 0.5.2. 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:
205
205
 
206
206
  | Platform | Configuration |
207
207
  |----------|---------------|
{nextrec-0.5.2 → nextrec-0.5.3}/docs/rtd/conf.py RENAMED
@@ -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.2"
14
+ release = "0.5.3"
15
15
 
16
16
  extensions = [
17
17
  "myst_parser",
nextrec-0.5.3/nextrec/__version__.py ADDED
@@ -0,0 +1 @@
1
+ __version__ = "0.5.3"
{nextrec-0.5.2 → nextrec-0.5.3}/nextrec/basic/model.py RENAMED
@@ -1707,7 +1707,7 @@ class BaseModel(SummarySet, FeatureSet, nn.Module):
1707
1707
  if (
1708
1708
  save_path is not None
1709
1709
  and not return_dataframe
1710
- and isinstance(data, (str, os.PathLike))
1710
+ and isinstance(data, (str, os.PathLike, DataLoader))
1711
1711
  ):
1712
1712
  if num_processes > 1 and not isinstance(data, (str, os.PathLike)):
1713
1713
  raise ValueError(
{nextrec-0.5.2 → nextrec-0.5.3}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "nextrec"
3
- version = "0.5.2"
3
+ version = "0.5.3"
4
4
  description = "A comprehensive recommendation library with match, ranking, and multi-task learning models"
5
5
  readme = "README.md"
6
6
  requires-python = ">=3.10"
nextrec-0.5.2/nextrec/__version__.py DELETED
@@ -1 +0,0 @@
1
- __version__ = "0.5.2"
{nextrec-0.5.2 → nextrec-0.5.3}/.gitignore RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/.readthedocs.yaml RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/CODE_OF_CONDUCT.md RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/CONTRIBUTING.md RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/LICENSE RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/MANIFEST.in RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/assets/logo.png RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/assets/test data.png RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/dataset/ctcvr_task.csv RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/dataset/match_task.csv RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/docs/rtd/Makefile RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/docs/rtd/index.md RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/docs/rtd/make.bat RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/docs/rtd/modules.rst RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/docs/rtd/nextrec.rst RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/nextrec/__init__.py RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/nextrec/basic/heads.py RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/nextrec/basic/layers.py RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/nextrec/cli.py RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/nextrec/utils/config.py RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/nextrec/utils/data.py RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/nextrec/utils/loss.py RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/nextrec/utils/model.py RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/nextrec/utils/types.py RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/pytest.ini RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/requirements.txt RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/scripts/format_code.py RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/test/__init__.py RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/test/conftest.py RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/test/helpers.py RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/test/run_tests.py RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/test/test_layers.py RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/test/test_losses.py RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/test/test_utils_data.py RENAMED
File without changes
{nextrec-0.5.2 → nextrec-0.5.3}/test_requirements.txt RENAMED
File without changes