nextrec 0.1.8__tar.gz → 0.1.11__tar.gz

{nextrec-0.1.8 → nextrec-0.1.11}/.gitignore

@@ -117,6 +117,7 @@ dmypy.json
 local_deepfm.py
 local_mmoe.py
 online_test.py
+linux_train_tutorial.py
 
 # DataProcessor saved files
 processor/

{nextrec-0.1.8 → nextrec-0.1.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextrec
-Version: 0.1.8
+Version: 0.1.11
 Summary: A comprehensive recommendation library with match, ranking, and multi-task learning models
 Project-URL: Homepage, https://github.com/zerolovesea/NextRec
 Project-URL: Repository, https://github.com/zerolovesea/NextRec
@@ -19,21 +19,26 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.10
-Requires-Dist: numpy<2.0,>=1.21; sys_platform == 'linux'
+Requires-Dist: numpy<2.0,>=1.21; sys_platform == 'linux' and python_version < '3.12'
+Requires-Dist: numpy<3.0,>=1.26; sys_platform == 'linux' and python_version >= '3.12'
 Requires-Dist: numpy>=1.23.0; sys_platform == 'win32'
 Requires-Dist: numpy>=1.24.0; sys_platform == 'darwin'
-Requires-Dist: pandas<2.0,>=1.5; sys_platform == 'linux'
+Requires-Dist: pandas<2.0,>=1.5; sys_platform == 'linux' and python_version < '3.12'
 Requires-Dist: pandas<2.3.0,>=2.1.0; sys_platform == 'win32'
 Requires-Dist: pandas>=2.0.0; sys_platform == 'darwin'
-Requires-Dist: pyarrow<13.0.0,>=10.0.0; sys_platform == 'linux'
+Requires-Dist: pandas>=2.1.0; sys_platform == 'linux' and python_version >= '3.12'
+Requires-Dist: pyarrow<13.0.0,>=10.0.0; sys_platform == 'linux' and python_version < '3.12'
 Requires-Dist: pyarrow<15.0.0,>=12.0.0; sys_platform == 'win32'
 Requires-Dist: pyarrow>=12.0.0; sys_platform == 'darwin'
-Requires-Dist: scikit-learn<2.0,>=1.2; sys_platform == 'linux'
+Requires-Dist: pyarrow>=16.0.0; sys_platform == 'linux' and python_version >= '3.12'
+Requires-Dist: scikit-learn<2.0,>=1.2; sys_platform == 'linux' and python_version < '3.12'
 Requires-Dist: scikit-learn>=1.3.0; sys_platform == 'darwin'
+Requires-Dist: scikit-learn>=1.3.0; sys_platform == 'linux' and python_version >= '3.12'
 Requires-Dist: scikit-learn>=1.3.0; sys_platform == 'win32'
-Requires-Dist: scipy<1.12,>=1.8; sys_platform == 'linux'
+Requires-Dist: scipy<1.12,>=1.8; sys_platform == 'linux' and python_version < '3.12'
 Requires-Dist: scipy>=1.10.0; sys_platform == 'darwin'
 Requires-Dist: scipy>=1.10.0; sys_platform == 'win32'
+Requires-Dist: scipy>=1.11.0; sys_platform == 'linux' and python_version >= '3.12'
 Requires-Dist: torch>=2.0.0
 Requires-Dist: torchvision>=0.15.0
 Requires-Dist: tqdm>=4.65.0
@@ -56,9 +61,9 @@ Description-Content-Type: text/markdown
 ![Python](https://img.shields.io/badge/Python-3.10+-blue.svg)
 ![PyTorch](https://img.shields.io/badge/PyTorch-1.10+-ee4c2c.svg)
 ![License](https://img.shields.io/badge/License-Apache%202.0-green.svg)
-![Version](https://img.shields.io/badge/Version-0.1.0-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.1.11-orange.svg)
 
-[中文版](README_zh.md)
+English | [中文版](README_zh.md)
 
 **A Unified, Efficient, and Scalable Recommendation System Framework**
 
@@ -264,7 +269,6 @@ NextRec is inspired by the following great open-source projects:
 - **torch-rechub** - A Lighting Pytorch Framework for Recommendation Models, Easy-to-use and Easy-to-extend.
 - **FuxiCTR** — Configurable and reproducible CTR prediction library  
 - **RecBole** — Unified and efficient recommendation library  
-- **PaddleRec** — Large-scale recommendation algorithm library  
 
 Special thanks to all open-source contributors!
 

{nextrec-0.1.8 → nextrec-0.1.11}/README.md

@@ -5,9 +5,9 @@
 ![Python](https://img.shields.io/badge/Python-3.10+-blue.svg)
 ![PyTorch](https://img.shields.io/badge/PyTorch-1.10+-ee4c2c.svg)
 ![License](https://img.shields.io/badge/License-Apache%202.0-green.svg)
-![Version](https://img.shields.io/badge/Version-0.1.0-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.1.11-orange.svg)
 
-[中文版](README_zh.md)
+English | [中文版](README_zh.md)
 
 **A Unified, Efficient, and Scalable Recommendation System Framework**
 
@@ -213,7 +213,6 @@ NextRec is inspired by the following great open-source projects:
 - **torch-rechub** - A Lighting Pytorch Framework for Recommendation Models, Easy-to-use and Easy-to-extend.
 - **FuxiCTR** — Configurable and reproducible CTR prediction library  
 - **RecBole** — Unified and efficient recommendation library  
-- **PaddleRec** — Large-scale recommendation algorithm library  
 
 Special thanks to all open-source contributors!
 

{nextrec-0.1.8 → nextrec-0.1.11}/README_zh.md

@@ -5,9 +5,9 @@
 ![Python](https://img.shields.io/badge/Python-3.10+-blue.svg)
 ![PyTorch](https://img.shields.io/badge/PyTorch-1.10+-ee4c2c.svg)
 ![License](https://img.shields.io/badge/License-Apache%202.0-green.svg)
-![Version](https://img.shields.io/badge/Version-0.1.0-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.1.11-orange.svg)
 
-[English Version](README.md)
+[English Version](README.md) | 中文版
 
 **统一、高效、可扩展的推荐系统框架**
 
@@ -210,7 +210,6 @@ NextRec 的开发受到以下优秀项目的启发：
 - [torch-rechub](https://github.com/datawhalechina/torch-rechub) - 灵活且易于扩展的推荐系统框架
 - [FuxiCTR](https://github.com/reczoo/FuxiCTR) - 可配置、可调优、可复现的 CTR 预测库
 - [RecBole](https://github.com/RUCAIBox/RecBole) - 统一、全面、高效的推荐库
-- [PaddleRec](https://github.com/PaddlePaddle/PaddleRec) - 大规模推荐算法库
 
 感谢开源社区的所有贡献者！
 

{nextrec-0.1.8 → nextrec-0.1.11}/docs/conf.py

@@ -12,7 +12,7 @@ sys.path.insert(0, os.path.abspath('../nextrec'))
 project = "NextRec"
 copyright = "2025, Yang Zhou"
 author = "Yang Zhou"
-release = "0.1.8"
+release = "0.1.11"
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration

nextrec-0.1.11/nextrec/__version__.py

@@ -0,0 +1 @@
+__version__ = "0.1.11"

{nextrec-0.1.8 → nextrec-0.1.11}/nextrec/basic/model.py

@@ -50,6 +50,7 @@ class BaseModel(nn.Module):
                  embedding_l2_reg: float = 0.0, 
                  dense_l2_reg: float = 0.0,
                  early_stop_patience: int = 20, 
+                 model_path: str = './',
                  model_id: str = 'baseline'): 
         
         super(BaseModel, self).__init__()
@@ -85,12 +86,12 @@ class BaseModel(nn.Module):
         self.early_stop_patience = early_stop_patience
         self._max_gradient_norm = 1.0   # Maximum gradient norm for gradient clipping
 
-        project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
         self.model_id = model_id
-        
-        checkpoint_dir = os.path.abspath(os.path.join(project_root, "..", "checkpoints"))
+
+        model_path = os.path.abspath(os.getcwd() if model_path in [None, './'] else model_path)
+        checkpoint_dir = os.path.join(model_path, "checkpoints", self.model_id)
         os.makedirs(checkpoint_dir, exist_ok=True)
-        self.checkpoint = os.path.join(checkpoint_dir, f"{self.model_name}_{self.model_id}_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}.model")
+        self.checkpoint = os.path.join(checkpoint_dir, f"{self.model_name}_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}.model")
         self.best = os.path.join(checkpoint_dir, f"{self.model_name}_{self.model_id}_best.model")
 
         self._logger_initialized = False

{nextrec-0.1.8 → nextrec-0.1.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "nextrec"
-version = "0.1.8"
+version = "0.1.11"
 description = "A comprehensive recommendation library with match, ranking, and multi-task learning models"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -38,12 +38,19 @@ dependencies = [
     "scikit-learn>=1.3.0; sys_platform == 'win32'",
     "pyarrow>=12.0.0,<15.0.0; sys_platform == 'win32'",
 
-    #linux
-    "numpy>=1.21,<2.0; sys_platform == 'linux'",
-    "pandas>=1.5,<2.0; sys_platform == 'linux'",
-    "scipy>=1.8,<1.12; sys_platform == 'linux'",
-    "scikit-learn>=1.2,<2.0; sys_platform == 'linux'",
-    "pyarrow>=10.0.0,<13.0.0; sys_platform == 'linux'",
+    # Linux, Python < 3.12
+    "numpy>=1.21,<2.0; sys_platform == 'linux' and python_version < '3.12'",
+    "pandas>=1.5,<2.0; sys_platform == 'linux' and python_version < '3.12'",
+    "scipy>=1.8,<1.12; sys_platform == 'linux' and python_version < '3.12'",
+    "scikit-learn>=1.2,<2.0; sys_platform == 'linux' and python_version < '3.12'",
+    "pyarrow>=10.0.0,<13.0.0; sys_platform == 'linux' and python_version < '3.12'",
+
+    # Linux, Python >= 3.12
+    "numpy>=1.26,<3.0; sys_platform == 'linux' and python_version >= '3.12'",
+    "pandas>=2.1.0; sys_platform == 'linux' and python_version >= '3.12'",
+    "scipy>=1.11.0; sys_platform == 'linux' and python_version >= '3.12'",
+    "scikit-learn>=1.3.0; sys_platform == 'linux' and python_version >= '3.12'",
+    "pyarrow>=16.0.0; sys_platform == 'linux' and python_version >= '3.12'",
 ]
 
 [project.optional-dependencies]

nextrec-0.1.8/nextrec/__version__.py

@@ -1 +0,0 @@
-__version__ = "0.1.8"