nextrec 0.4.18__tar.gz → 0.4.19__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextrec
-Version: 0.4.18
+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
@@ -67,8 +67,8 @@ 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.4.18-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)
 
@@ -243,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.18，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
+预测结果固定保存到 `{checkpoint_path}/predictions/{name}.{save_data_format}`。
+
+> 截止当前版本0.4.19，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
 
 ## 兼容平台
 
-当前最新版本为0.4.18，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
+当前最新版本为0.4.19，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
 
 | 平台 | 配置 | 
 |------|------|

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

@@ -8,8 +8,8 @@
 ![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.18-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)
 
@@ -184,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.18，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
+预测结果固定保存到 `{checkpoint_path}/predictions/{name}.{save_data_format}`。
+
+> 截止当前版本0.4.19，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
 
 ## 兼容平台
 
-当前最新版本为0.4.18，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
+当前最新版本为0.4.19，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
 
 | 平台 | 配置 | 
 |------|------|

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

@@ -8,7 +8,8 @@
 ![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.18-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)
 
@@ -187,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.18, 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.18. 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.18 → 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.18"
+release = "0.4.19"
 
 extensions = [
     "myst_parser",

nextrec-0.4.19/nextrec/__version__.py

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

{nextrec-0.4.18 → nextrec-0.4.19}/nextrec/cli.py

@@ -585,13 +585,13 @@ def predict_model(predict_config_path: str) -> None:
         chunk_size=predict_cfg.get("chunk_size", 20000),
     )
 
-    # Build output path: {checkpoint_path}/predictions/{name}.{save_data_format}
+
     save_format = predict_cfg.get(
         "save_data_format", predict_cfg.get("save_format", "csv")
     )
     pred_name = predict_cfg.get("name", "pred")
-    # Pass filename with extension to let model.predict handle path resolution
-    save_path = f"{pred_name}.{save_format}"
+
+    save_path = checkpoint_base / "predictions" / f"{pred_name}.{save_format}"
 
     start = time.time()
     logger.info("")
@@ -600,7 +600,7 @@ def predict_model(predict_config_path: str) -> None:
         batch_size=batch_size,
         include_ids=bool(id_columns),
         return_dataframe=False,
-        save_path=save_path,
+        save_path=str(save_path),
         save_format=save_format,
         num_workers=predict_cfg.get("num_workers", 0),
     )

{nextrec-0.4.18 → nextrec-0.4.19}/nextrec_cli_preset/NextRec-CLI.md

@@ -84,7 +84,7 @@ dataloader:
   valid_batch_size: 512                 # Validation batch size
   valid_shuffle: false                  # Shuffle validation data
   num_workers: 4                        # Number of parallel data loading workers
-                                       # 0 = single process, >0 = multiprocessing
+                                        # 0 = single process, >0 = multiprocessing
   # chunk_size: 20000                   # Chunk size when streaming=true
 
 train:
@@ -187,7 +187,7 @@ The prediction configuration file defines the model inference pipeline.
 #### Configuration Structure
 
 ```yaml
-checkpoint_path: /path/to/checkpoint/directory  # Required checkpoint directory
+checkpoint_path: /path/to/checkpoint/directory   # Required checkpoint directory
 # model_config: /path/to/model_config.yaml       # Optional model config override
 # session:
 #   id: my_experiment_session                    # Optional session ID
@@ -195,10 +195,10 @@ checkpoint_path: /path/to/checkpoint/directory  # Required checkpoint directory
 predict:
   data_path: /path/to/prediction/data   # Prediction data path
   source_data_format: parquet           # Input data format: csv, parquet, feather
-                                       # Use 'auto' for automatic detection
+                                        # Use 'auto' for automatic detection
   id_column: user_id                    # ID column name (optional, for linking predictions)
   name: pred                            # Output filename (without extension)
-                                       # Final path: {checkpoint_path}/predictions/{name}.{save_data_format}
+                                        # Final path: {checkpoint_path}/predictions/{name}.{save_data_format}
   save_data_format: csv                 # Output format: csv, parquet, feather
   preview_rows: 5                       # Number of preview rows (output to log)
   batch_size: 512                       # Prediction batch size

{nextrec-0.4.18 → nextrec-0.4.19}/nextrec_cli_preset/NextRec-CLI_zh.md

@@ -84,7 +84,7 @@ dataloader:
   valid_batch_size: 512                 # 验证批次大小
   valid_shuffle: false                  # 是否打乱验证数据
   num_workers: 4                        # 数据加载并行进程数
-                                       # 0=单进程, >0=多进程
+                                        # 0=单进程, >0=多进程
   # chunk_size: 20000                   # streaming=true 时的分块大小
 
 train:
@@ -195,10 +195,10 @@ checkpoint_path: /path/to/checkpoint/directory  # 必填 checkpoint 目录
 predict:
   data_path: /path/to/prediction/data   # 预测数据路径
   source_data_format: parquet           # 输入数据格式：csv, parquet, feather
-                                       # 使用 auto 自动识别
+                                        # 使用 auto 自动识别
   id_column: user_id                    # ID列名（可选，用于关联预测结果）
   name: pred                            # 输出文件名（不含扩展名）
-                                       # 最终路径: {checkpoint_path}/predictions/{name}.{save_data_format}
+                                        # 最终路径: {checkpoint_path}/predictions/{name}.{save_data_format}
   save_data_format: csv                 # 输出格式：csv, parquet, feather
   preview_rows: 5                       # 预览行数（输出到日志）
   batch_size: 512                       # 预测批次大小
@@ -380,8 +380,8 @@ model: deepfm
 params:
   mlp_params:
     dims: [256, 128, 64]               # MLP 层维度
-    activation: relu                    # 激活函数
-    dropout: 0.3                        # Dropout 比率
+    activation: relu                   # 激活函数
+    dropout: 0.3                       # Dropout 比率
   embedding_l2_reg: 1.0e-5             # Embedding L2 正则化
   dense_l2_reg: 1.0e-4                 # Dense 层 L2 正则化
 ```
@@ -430,7 +430,7 @@ params:
 ```yaml
 model: masknet
 params:
-  architecture: parallel                 # 模型类型：parallel, serial
+  architecture: parallel               # 模型类型：parallel, serial
   num_blocks: 3                        # Block 数量
   mask_hidden_dim: 64                  # Mask 隐藏层维度
   block_hidden_dim: 256                # Block 隐藏层维度

{nextrec-0.4.18 → nextrec-0.4.19}/pyproject.toml

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

nextrec-0.4.18/nextrec/__version__.py

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