nextrec 0.4.27__tar.gz → 0.4.28__tar.gz

{nextrec-0.4.27 → nextrec-0.4.28}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextrec
-Version: 0.4.27
+Version: 0.4.28
 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
@@ -69,7 +69,7 @@ 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.27-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.4.28-orange.svg)
 [![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
 
 中文文档 | [English Version](README_en.md)
@@ -102,7 +102,7 @@ NextRec是一个基于PyTorch的现代推荐系统框架，旨在为研究工程
 - **高效训练与评估**：内置多种优化器、学习率调度、早停、模型检查点与详细的日志管理，开箱即用。
 
 ## NextRec近期进展
-- **01/01/2026** 新年好，在v0.4.27中加入了多个多目标模型的支持：[APG](nextrec/models/multi_task/apg.py), [ESCM](nextrec/models/multi_task/escm.py), [HMoE](nextrec/models/multi_task/hmoe.py), [Cross Stitch](nextrec/models/multi_task/cross_stitch.py)
+- **01/01/2026** 新年好，在v0.4.28中加入了多个多目标模型的支持：[APG](nextrec/models/multi_task/apg.py), [ESCM](nextrec/models/multi_task/escm.py), [HMoE](nextrec/models/multi_task/hmoe.py), [Cross Stitch](nextrec/models/multi_task/cross_stitch.py)
 - **28/12/2025** 在v0.4.21中加入了对SwanLab和Wandb的支持，通过model的`fit`方法进行配置：`use_swanlab=True, swanlab_kwargs={"project": "NextRec","name":"tutorial_movielens_deepfm"},`
 - **21/12/2025** 在v0.4.16中加入了对[GradNorm](/nextrec/loss/grad_norm.py)的支持，通过compile的`loss_weight='grad_norm'`进行配置
 - **12/12/2025** 在v0.4.9中加入了[RQ-VAE](/nextrec/models/representation/rqvae.py)模块。配套的[数据集](/dataset/ecommerce_task.csv)和[代码](tutorials/notebooks/zh/使用RQ-VAE构建语义ID.ipynb)已经同步在仓库中
@@ -254,11 +254,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
 
 预测结果固定保存到 `{checkpoint_path}/predictions/{name}.{save_data_format}`。
 
-> 截止当前版本0.4.27，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
+> 截止当前版本0.4.28，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
 
 ## 兼容平台
 
-当前最新版本为0.4.27，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
+当前最新版本为0.4.28，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
 
 | 平台 | 配置 | 
 |------|------|

{nextrec-0.4.27 → nextrec-0.4.28}/README.md

@@ -8,7 +8,7 @@
 ![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.27-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.4.28-orange.svg)
 [![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
 
 中文文档 | [English Version](README_en.md)
@@ -41,7 +41,7 @@ NextRec是一个基于PyTorch的现代推荐系统框架，旨在为研究工程
 - **高效训练与评估**：内置多种优化器、学习率调度、早停、模型检查点与详细的日志管理，开箱即用。
 
 ## NextRec近期进展
-- **01/01/2026** 新年好，在v0.4.27中加入了多个多目标模型的支持：[APG](nextrec/models/multi_task/apg.py), [ESCM](nextrec/models/multi_task/escm.py), [HMoE](nextrec/models/multi_task/hmoe.py), [Cross Stitch](nextrec/models/multi_task/cross_stitch.py)
+- **01/01/2026** 新年好，在v0.4.28中加入了多个多目标模型的支持：[APG](nextrec/models/multi_task/apg.py), [ESCM](nextrec/models/multi_task/escm.py), [HMoE](nextrec/models/multi_task/hmoe.py), [Cross Stitch](nextrec/models/multi_task/cross_stitch.py)
 - **28/12/2025** 在v0.4.21中加入了对SwanLab和Wandb的支持，通过model的`fit`方法进行配置：`use_swanlab=True, swanlab_kwargs={"project": "NextRec","name":"tutorial_movielens_deepfm"},`
 - **21/12/2025** 在v0.4.16中加入了对[GradNorm](/nextrec/loss/grad_norm.py)的支持，通过compile的`loss_weight='grad_norm'`进行配置
 - **12/12/2025** 在v0.4.9中加入了[RQ-VAE](/nextrec/models/representation/rqvae.py)模块。配套的[数据集](/dataset/ecommerce_task.csv)和[代码](tutorials/notebooks/zh/使用RQ-VAE构建语义ID.ipynb)已经同步在仓库中
@@ -193,11 +193,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
 
 预测结果固定保存到 `{checkpoint_path}/predictions/{name}.{save_data_format}`。
 
-> 截止当前版本0.4.27，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
+> 截止当前版本0.4.28，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
 
 ## 兼容平台
 
-当前最新版本为0.4.27，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
+当前最新版本为0.4.28，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
 
 | 平台 | 配置 | 
 |------|------|

{nextrec-0.4.27 → nextrec-0.4.28}/README_en.md

@@ -8,7 +8,7 @@
 ![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.27-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.4.28-orange.svg)
 [![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
 
 English | [中文文档](README.md)
@@ -44,7 +44,7 @@ NextRec is a modern recommendation framework built on PyTorch, delivering a unif
 
 ## NextRec Progress
 
-- **01/01/2026** Happy New Year! In v0.4.27, added support for multiple multi-task models: [APG](/nextrec/models/multi_task/apg.py), [ESCM](/nextrec/models/multi_task/escm.py), [HMoE](/nextrec/models/multi_task/hmoe.py), [Cross Stitch](/nextrec/models/multi_task/cross_stitch.py)
+- **01/01/2026** Happy New Year! In v0.4.28, added support for multiple multi-task models: [APG](/nextrec/models/multi_task/apg.py), [ESCM](/nextrec/models/multi_task/escm.py), [HMoE](/nextrec/models/multi_task/hmoe.py), [Cross Stitch](/nextrec/models/multi_task/cross_stitch.py)
 - **28/12/2025** Added support for SwanLab and Weights & Biases in v0.4.21, configurable via the model `fit` method: `use_swanlab=True, swanlab_kwargs={"project": "NextRec","name":"tutorial_movielens_deepfm"},`
 - **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
 - **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.
@@ -196,11 +196,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
 
 Prediction outputs are saved under `{checkpoint_path}/predictions/{name}.{save_data_format}`.
 
-> As of version 0.4.27, NextRec CLI supports single-machine training; distributed training features are currently under development.
+> As of version 0.4.28, NextRec CLI supports single-machine training; distributed training features are currently under development.
 
 ## Platform Compatibility
 
-The current version is 0.4.27. 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.28. 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.27 → nextrec-0.4.28}/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.27"
+release = "0.4.28"
 
 extensions = [
     "myst_parser",

nextrec-0.4.28/nextrec/__version__.py

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

{nextrec-0.4.27 → nextrec-0.4.28}/nextrec/data/batch_utils.py

@@ -5,13 +5,27 @@ Date: create on 03/12/2025
 Author: Yang Zhou, zyaztec@gmail.com
 """
 
-from typing import Any, Mapping
+from typing import Any, Mapping, Literal
 
 import numpy as np
 import torch
 
 
-def stack_section(batch: list[dict], section: str):
+def stack_section(batch: list[dict], section: Literal["features", "labels", "ids"]):
+    """   
+    input example:
+    batch = [
+        {"features": {"f1": tensor1, "f2": tensor2}, "labels": {"label": tensor3}},
+        {"features": {"f1": tensor4, "f2": tensor5}, "labels": {"label": tensor6}},
+        ...
+    ]
+    output example:
+    {
+        "f1": torch.stack([tensor1, tensor4], dim=0),
+        "f2": torch.stack([tensor2, tensor5], dim=0),
+    }
+    
+    """
     entries = [item.get(section) for item in batch if item.get(section) is not None]
     if not entries:
         return None
@@ -22,7 +36,13 @@ def stack_section(batch: list[dict], section: str):
             for item in batch
             if item.get(section) is not None and name in item[section]
         ]
-        merged[name] = torch.stack(tensors, dim=0)
+        tensor_sample = tensors[0]
+        if isinstance(tensor_sample, torch.Tensor):
+            merged[name] = torch.stack(tensors, dim=0)
+        elif isinstance(tensor_sample, np.ndarray):
+            merged[name] = np.stack(tensors, axis=0)
+        else:
+            merged[name] = tensors
     return merged
 
 

{nextrec-0.4.27 → nextrec-0.4.28}/nextrec/data/dataloader.py

@@ -2,7 +2,7 @@
 Dataloader definitions
 
 Date: create on 27/10/2025
-Checkpoint: edit on 24/12/2025
+Checkpoint: edit on 01/01/2026
 Author: Yang Zhou,zyaztec@gmail.com
 """
 
@@ -523,13 +523,8 @@ def build_tensors_from_data(
                 raise KeyError(
                     f"[RecDataLoader Error] ID column '{id_col}' not found in provided data."
                 )
-            try:
-                id_arr = np.asarray(column, dtype=np.int64)
-            except Exception as exc:
-                raise TypeError(
-                    f"[RecDataLoader Error] ID column '{id_col}' must contain numeric values. Received dtype={np.asarray(column).dtype}, error: {exc}"
-                ) from exc
-            id_tensors[id_col] = to_tensor(id_arr, dtype=torch.long)
+            # Normalize all id columns to strings for consistent downstream handling.
+            id_tensors[id_col] = np.asarray(column, dtype=str)
     if not feature_tensors:
         return None
     return {"features": feature_tensors, "labels": label_tensors, "ids": id_tensors}

{nextrec-0.4.27 → nextrec-0.4.28}/nextrec/models/multi_task/pepnet.py

@@ -61,9 +61,9 @@ from nextrec.utils.model import select_features
 from nextrec.utils.types import TaskTypeName
 
 
-class PPNetBlock(nn.Module):
+class PPNet(nn.Module):
     """
-    PEPNet block with per-layer gates conditioned on task context.
+    PPNet: per-task tower with layer-wise gates conditioned on task context.
     """
 
     def __init__(
@@ -274,18 +274,21 @@ class PEPNet(BaseModel):
         )
         task_dim = domain_dim + user_dim + item_dim
 
-        self.feature_gate = GateMLP(
+        # EPNet: shared feature-level gate (paper's EPNet).
+        self.epnet = GateMLP(
             input_dim=input_dim + domain_dim,
             hidden_dim=feature_gate_mlp_params["hidden_dim"],
             output_dim=input_dim,
             activation=feature_gate_mlp_params["activation"],
             dropout=feature_gate_mlp_params["dropout"],
             use_bn=feature_gate_mlp_params["use_bn"],
+            scale_factor=2.0,
         )
 
-        self.ppn_blocks = nn.ModuleList(
+        # PPNet: per-task gated towers (paper's PPNet).
+        self.ppnet_blocks = nn.ModuleList(
             [
-                PPNetBlock(
+                PPNet(
                     input_dim=input_dim,
                     output_dim=1,
                     gate_input_dim=input_dim + task_dim,
@@ -300,9 +303,9 @@ class PEPNet(BaseModel):
         self.prediction_layer = TaskHead(
             task_type=self.task, task_dims=[1] * self.nums_task
         )
-        self.grad_norm_shared_modules = ["embedding", "feature_gate"]
+        self.grad_norm_shared_modules = ["embedding", "epnet"]
         self.register_regularization_weights(
-            embedding_attr="embedding", include_modules=["feature_gate", "ppn_blocks"]
+            embedding_attr="embedding", include_modules=["epnet", "ppnet_blocks"]
         )
 
     def forward(self, x: dict[str, torch.Tensor]) -> torch.Tensor:
@@ -327,11 +330,11 @@ class PEPNet(BaseModel):
         task_sf_emb = torch.cat(task_parts, dim=-1)
 
         gate_input = torch.cat([dnn_input.detach(), domain_emb], dim=-1)
-        dnn_input = self.feature_gate(gate_input) * dnn_input
+        dnn_input = self.epnet(gate_input) * dnn_input
 
         task_logits = []
-        for block in self.ppn_blocks:
-            task_logits.append(block(o_ep=dnn_input, o_prior=task_sf_emb))
+        for block in self.ppnet_blocks:
+             task_logits.append(block(o_ep=dnn_input, o_prior=task_sf_emb))
 
         y = torch.cat(task_logits, dim=1)
         return self.prediction_layer(y)

{nextrec-0.4.27 → nextrec-0.4.28}/pyproject.toml

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

nextrec-0.4.27/nextrec/__version__.py

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