nextrec 0.1.10__tar.gz → 0.1.11__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextrec
-Version: 0.1.10
+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
@@ -61,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**
 
@@ -269,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.10 → 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.10 → 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.10 → 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.10"
+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.10 → 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.10 → nextrec-0.1.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "nextrec"
-version = "0.1.10"
+version = "0.1.11"
 description = "A comprehensive recommendation library with match, ranking, and multi-task learning models"
 readme = "README.md"
 requires-python = ">=3.10"

nextrec-0.1.10/nextrec/__version__.py

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