nextrec 0.5.0__tar.gz → 0.5.2__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.
- {nextrec-0.5.0 → nextrec-0.5.2}/PKG-INFO +4 -5
- {nextrec-0.5.0 → nextrec-0.5.2}/README.md +3 -4
- {nextrec-0.5.0 → nextrec-0.5.2}/README_en.md +3 -4
- {nextrec-0.5.0 → nextrec-0.5.2}/docs/rtd/conf.py +1 -1
- nextrec-0.5.2/nextrec/__version__.py +1 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/basic/model.py +288 -181
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/basic/summary.py +21 -4
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/cli.py +36 -17
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/data/batch_utils.py +1 -1
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/data/data_processing.py +1 -35
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/data/data_utils.py +0 -4
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/data/dataloader.py +125 -103
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/data/preprocessor.py +141 -92
- nextrec-0.5.2/nextrec/models/multi_task/__init__.py +0 -0
- nextrec-0.5.2/nextrec/models/ranking/__init__.py +0 -0
- nextrec-0.5.2/nextrec/models/retrieval/__init__.py +0 -0
- nextrec-0.5.2/nextrec/models/tree_base/__init__.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/tree_base/base.py +14 -23
- nextrec-0.5.2/nextrec/utils/__init__.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/utils/data.py +39 -119
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/utils/model.py +5 -14
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/utils/torch_utils.py +6 -1
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/NextRec-CLI.md +8 -5
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/NextRec-CLI_zh.md +8 -5
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/predict_config.yaml +5 -2
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/predict_config_template.yaml +5 -2
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/train_config_template.yaml +2 -2
- {nextrec-0.5.0 → nextrec-0.5.2}/pyproject.toml +1 -1
- nextrec-0.5.2/scripts/profile_predict.py +521 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/test/test_losses.py +3 -6
- {nextrec-0.5.0 → nextrec-0.5.2}/test/test_preprocessor.py +0 -31
- {nextrec-0.5.0 → nextrec-0.5.2}/test/test_utils_data.py +5 -52
- {nextrec-0.5.0 → nextrec-0.5.2}/tutorials/distributed/example_distributed_training.py +1 -1
- {nextrec-0.5.0 → nextrec-0.5.2}/tutorials/example_match.py +2 -2
- {nextrec-0.5.0 → nextrec-0.5.2}/tutorials/example_multitask.py +1 -1
- {nextrec-0.5.0 → nextrec-0.5.2}/tutorials/movielen_match_dssm.py +2 -3
- {nextrec-0.5.0 → nextrec-0.5.2}/tutorials/run_all_match_models.py +2 -1
- {nextrec-0.5.0 → nextrec-0.5.2}/tutorials/run_all_multitask_models.py +1 -2
- {nextrec-0.5.0 → nextrec-0.5.2}/tutorials/run_all_ranking_models.py +1 -1
- nextrec-0.5.0/nextrec/__version__.py +0 -1
- nextrec-0.5.0/nextrec/data/__init__.py +0 -52
- nextrec-0.5.0/nextrec/loss/__init__.py +0 -36
- nextrec-0.5.0/nextrec/models/generative/__init__.py +0 -9
- nextrec-0.5.0/nextrec/models/tree_base/__init__.py +0 -15
- nextrec-0.5.0/nextrec/utils/__init__.py +0 -119
- {nextrec-0.5.0 → nextrec-0.5.2}/.github/workflows/publish.yml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/.github/workflows/tests.yml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/.gitignore +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/.readthedocs.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/CODE_OF_CONDUCT.md +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/CONTRIBUTING.md +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/LICENSE +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/MANIFEST.in +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/assets/Feature Configuration.png +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/assets/Model Parameters.png +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/assets/Training Configuration.png +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/assets/Training logs.png +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/assets/logo.png +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/assets/mmoe_tutorial.png +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/assets/nextrec_diagram.png +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/assets/test data.png +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/dataset/ctcvr_task.csv +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/dataset/ecommerce_task.csv +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/dataset/match_task.csv +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/dataset/movielens_100k.csv +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/dataset/multitask_task.csv +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/dataset/ranking_task.csv +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/docs/en/Getting started guide.md +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/docs/rtd/Makefile +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/docs/rtd/index.md +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/docs/rtd/make.bat +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/docs/rtd/modules.rst +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/docs/rtd/nextrec.basic.rst +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/docs/rtd/nextrec.data.rst +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/docs/rtd/nextrec.loss.rst +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/docs/rtd/nextrec.rst +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/docs/rtd/nextrec.utils.rst +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/docs/rtd/requirements.txt +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/docs/zh//345/277/253/351/200/237/344/270/212/346/211/213.md" +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/__init__.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/basic/__init__.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/basic/activation.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/basic/asserts.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/basic/callback.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/basic/features.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/basic/heads.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/basic/layers.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/basic/loggers.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/basic/metrics.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/basic/session.py +0 -0
- {nextrec-0.5.0/nextrec/models/multi_task → nextrec-0.5.2/nextrec/data}/__init__.py +0 -0
- {nextrec-0.5.0/nextrec/models/ranking → nextrec-0.5.2/nextrec/loss}/__init__.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/loss/grad_norm.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/loss/listwise.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/loss/pairwise.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/loss/pointwise.py +0 -0
- {nextrec-0.5.0/nextrec/models/retrieval → nextrec-0.5.2/nextrec/models/generative}/__init__.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/generative/tiger.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/multi_task/[pre]aitm.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/multi_task/[pre]snr_trans.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/multi_task/apg.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/multi_task/cross_stitch.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/multi_task/escm.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/multi_task/esmm.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/multi_task/hmoe.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/multi_task/mmoe.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/multi_task/pepnet.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/multi_task/ple.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/multi_task/poso.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/multi_task/share_bottom.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/ranking/afm.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/ranking/autoint.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/ranking/dcn.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/ranking/dcn_v2.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/ranking/deepfm.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/ranking/dien.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/ranking/din.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/ranking/eulernet.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/ranking/ffm.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/ranking/fibinet.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/ranking/fm.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/ranking/lr.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/ranking/masknet.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/ranking/pnn.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/ranking/widedeep.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/ranking/xdeepfm.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/representation/__init__.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/representation/rqvae.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/retrieval/dssm.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/retrieval/dssm_v2.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/retrieval/mind.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/retrieval/sdm.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/retrieval/youtube_dnn.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/sequential/hstu.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/tree_base/catboost.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/tree_base/lightgbm.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/models/tree_base/xgboost.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/utils/config.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/utils/console.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/utils/embedding.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/utils/loss.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/utils/onnx_utils.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec/utils/types.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/feature_config.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/afm.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/apg.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/autoint.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/cross_stitch.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/dcn.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/deepfm.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/din.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/escm.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/esmm.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/fibinet.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/fm.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/hmoe.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/masknet.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/mmoe.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/pepnet.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/ple.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/pnn.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/poso.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/share_bottom.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/widedeep.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/model_configs/xdeepfm.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/nextrec_cli_preset/train_config.yaml +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/pytest.ini +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/requirements.txt +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/scripts/format_code.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/test/__init__.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/test/conftest.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/test/helpers.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/test/run_tests.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/test/test_base_model_regularization.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/test/test_generative_models.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/test/test_layers.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/test/test_match_models.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/test/test_multitask_models.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/test/test_onnx_models.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/test/test_ranking_models.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/test/test_utils_console.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/test/test_utils_embedding.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/test_requirements.txt +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/tutorials/distributed/example_distributed_training_large_dataset.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/tutorials/example_onnx.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/tutorials/example_ranking_din.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/tutorials/example_tree.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/tutorials/movielen_ranking_deepfm.py +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/tutorials/notebooks/en/Build semantic ID with RQ-VAE.ipynb +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/tutorials/notebooks/en/Hands on dataprocessor.ipynb +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/tutorials/notebooks/en/Hands on nextrec.ipynb +0 -0
- {nextrec-0.5.0 → nextrec-0.5.2}/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
- {nextrec-0.5.0 → nextrec-0.5.2}/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
- {nextrec-0.5.0 → nextrec-0.5.2}/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.
|
|
3
|
+
Version: 0.5.2
|
|
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
|

|
|
74
74
|

|
|
75
75
|

|
|
76
|
-

|
|
77
77
|
[](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.
|
|
262
|
+
> 截止当前版本0.5.2,NextRec CLI支持单机训练,分布式训练相关功能尚在开发中。
|
|
264
263
|
|
|
265
264
|
## 兼容平台
|
|
266
265
|
|
|
267
|
-
当前最新版本为0.5.
|
|
266
|
+
当前最新版本为0.5.2,所有模型和测试代码均已在以下平台通过验证,如果开发者在使用中遇到兼容问题,请在issue区提出错误报告及系统版本:
|
|
268
267
|
|
|
269
268
|
| 平台 | 配置 |
|
|
270
269
|
|------|------|
|
|
@@ -8,7 +8,7 @@
|
|
|
8
8
|

|
|
9
9
|

|
|
10
10
|

|
|
11
|
-

|
|
12
12
|
[](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.
|
|
197
|
+
> 截止当前版本0.5.2,NextRec CLI支持单机训练,分布式训练相关功能尚在开发中。
|
|
199
198
|
|
|
200
199
|
## 兼容平台
|
|
201
200
|
|
|
202
|
-
当前最新版本为0.5.
|
|
201
|
+
当前最新版本为0.5.2,所有模型和测试代码均已在以下平台通过验证,如果开发者在使用中遇到兼容问题,请在issue区提出错误报告及系统版本:
|
|
203
202
|
|
|
204
203
|
| 平台 | 配置 |
|
|
205
204
|
|------|------|
|
|
@@ -8,7 +8,7 @@
|
|
|
8
8
|

|
|
9
9
|

|
|
10
10
|

|
|
11
|
-

|
|
12
12
|
[](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.
|
|
200
|
+
> As of version 0.5.2, 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.
|
|
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:
|
|
206
205
|
|
|
207
206
|
| Platform | Configuration |
|
|
208
207
|
|----------|---------------|
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
__version__ = "0.5.2"
|