nextrec 0.4.17__tar.gz → 0.4.19__tar.gz

{nextrec-0.4.17 → nextrec-0.4.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextrec
-Version: 0.4.17
+Version: 0.4.19
 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
@@ -63,13 +63,12 @@ Description-Content-Type: text/markdown
 
 <div align="center">
 
-[![PyPI Downloads](https://static.pepy.tech/personalized-badge/nextrec?period=total&units=NONE&left_color=BLACK&right_color=GREEN&left_text=PyPI-downloads)](https://pepy.tech/projects/nextrec)
+[![PyPI Downloads](https://static.pepy.tech/personalized-badge/nextrec?period=total&units=NONE&left_color=BLACK&right_color=GREEN&left_text=PyPI-downloads)](https://pypistats.org/packages/nextrec)
 ![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.4.17-orange.svg)
-
+![Version](https://img.shields.io/badge/Version-0.4.19-orange.svg)
+[![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
 
 中文文档 | [English Version](README_en.md)
 
@@ -244,11 +243,13 @@ nextrec --mode=train --train_config=path/to/train_config.yaml
 nextrec --mode=predict --predict_config=path/to/predict_config.yaml
 ```
 
-> 截止当前版本0.4.17，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
+预测结果固定保存到 `{checkpoint_path}/predictions/{name}.{save_data_format}`。
+
+> 截止当前版本0.4.19，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
 
 ## 兼容平台
 
-当前最新版本为0.4.17，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
+当前最新版本为0.4.19，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
 
 | 平台 | 配置 | 
 |------|------|

{nextrec-0.4.17 → nextrec-0.4.19}/README.md

@@ -4,13 +4,12 @@
 
 <div align="center">
 
-[![PyPI Downloads](https://static.pepy.tech/personalized-badge/nextrec?period=total&units=NONE&left_color=BLACK&right_color=GREEN&left_text=PyPI-downloads)](https://pepy.tech/projects/nextrec)
+[![PyPI Downloads](https://static.pepy.tech/personalized-badge/nextrec?period=total&units=NONE&left_color=BLACK&right_color=GREEN&left_text=PyPI-downloads)](https://pypistats.org/packages/nextrec)
 ![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.4.17-orange.svg)
-
+![Version](https://img.shields.io/badge/Version-0.4.19-orange.svg)
+[![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
 
 中文文档 | [English Version](README_en.md)
 
@@ -185,11 +184,13 @@ nextrec --mode=train --train_config=path/to/train_config.yaml
 nextrec --mode=predict --predict_config=path/to/predict_config.yaml
 ```
 
-> 截止当前版本0.4.17，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
+预测结果固定保存到 `{checkpoint_path}/predictions/{name}.{save_data_format}`。
+
+> 截止当前版本0.4.19，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
 
 ## 兼容平台
 
-当前最新版本为0.4.17，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
+当前最新版本为0.4.19，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
 
 | 平台 | 配置 | 
 |------|------|

{nextrec-0.4.17 → nextrec-0.4.19}/README_en.md

@@ -4,12 +4,12 @@
 
 <div align="center">
 
-[![PyPI Downloads](https://static.pepy.tech/personalized-badge/nextrec?period=total&units=NONE&left_color=BLACK&right_color=GREEN&left_text=PyPI-downloads)](https://pepy.tech/projects/nextrec)
+[![PyPI Downloads](https://static.pepy.tech/personalized-badge/nextrec?period=total&units=NONE&left_color=BLACK&right_color=GREEN&left_text=PyPI-downloads)](https://pypistats.org/packages/nextrec)
 ![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.4.17-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.4.19-orange.svg)
+[![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
 
 English | [中文文档](README.md)
 
@@ -188,11 +188,13 @@ nextrec --mode=train --train_config=path/to/train_config.yaml
 nextrec --mode=predict --predict_config=path/to/predict_config.yaml
 ```
 
-> As of version 0.4.17, NextRec CLI supports single-machine training; distributed training features are currently under development.
+Prediction outputs are saved under `{checkpoint_path}/predictions/{name}.{save_data_format}`.
+
+> As of version 0.4.19, NextRec CLI supports single-machine training; distributed training features are currently under development.
 
 ## Platform Compatibility
 
-The current version is 0.4.17. 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:
+The current version is 0.4.19. 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:
 
 | Platform | Configuration | 
 |----------|---------------|

{nextrec-0.4.17 → nextrec-0.4.19}/docs/rtd/conf.py

@@ -11,7 +11,7 @@ sys.path.insert(0, str(PROJECT_ROOT / "nextrec"))
 project = "NextRec"
 copyright = "2025, Yang Zhou"
 author = "Yang Zhou"
-release = "0.4.17"
+release = "0.4.19"
 
 extensions = [
     "myst_parser",

nextrec-0.4.19/nextrec/__version__.py

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

{nextrec-0.4.17 → nextrec-0.4.19}/nextrec/basic/heads.py

@@ -91,9 +91,7 @@ class RetrievalHead(nn.Module):
             elif self.similarity_metric == "euclidean":
                 similarity = -torch.sum((user_emb - item_emb) ** 2, dim=-1)
             else:
-                raise ValueError(
-                    f"Unknown similarity metric: {self.similarity_metric}"
-                )
+                raise ValueError(f"Unknown similarity metric: {self.similarity_metric}")
 
             similarity = similarity / self.temperature
         if self.training_mode == "pointwise" and self.apply_sigmoid:

{nextrec-0.4.17 → nextrec-0.4.19}/nextrec/basic/loggers.py

@@ -2,7 +2,7 @@
 NextRec Basic Loggers
 
 Date: create on 27/10/2025
-Checkpoint: edit on 20/12/2025
+Checkpoint: edit on 24/12/2025
 Author: Yang Zhou, zyaztec@gmail.com
 """
 
@@ -148,18 +148,18 @@ class TrainingLogger:
     def __init__(
         self,
         session: Session,
-        enable_tensorboard: bool,
+        use_tensorboard: bool,
         log_name: str = "training_metrics.jsonl",
     ) -> None:
         self.session = session
-        self.enable_tensorboard = enable_tensorboard
+        self.use_tensorboard = use_tensorboard
         self.log_path = session.metrics_dir / log_name
         self.log_path.parent.mkdir(parents=True, exist_ok=True)
 
         self.tb_writer = None
         self.tb_dir = None
 
-        if self.enable_tensorboard:
+        if self.use_tensorboard:
             self._init_tensorboard()
 
     def _init_tensorboard(self) -> None:
@@ -169,7 +169,7 @@ class TrainingLogger:
             logging.warning(
                 "[TrainingLogger] tensorboard not installed, disable tensorboard logging."
             )
-            self.enable_tensorboard = False
+            self.use_tensorboard = False
             return
         tb_dir = self.session.logs_dir / "tensorboard"
         tb_dir.mkdir(parents=True, exist_ok=True)