nextrec 0.4.27__py3-none-any.whl → 0.4.28__py3-none-any.whl

nextrec/__version__.py

@@ -1 +1 @@
-__version__ = "0.4.27"
+__version__ = "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/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/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.dist-info → nextrec-0.4.28.dist-info}/METADATA

@@ -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.dist-info → nextrec-0.4.28.dist-info}/RECORD

@@ -1,5 +1,5 @@
 nextrec/__init__.py,sha256=_M3oUqyuvQ5k8Th_3wId6hQ_caclh7M5ad51XN09m98,235
-nextrec/__version__.py,sha256=M6cnWFk1Yz0fBKyJSOiFIvowDDGt2oz9dCYCtlYnCj4,23
+nextrec/__version__.py,sha256=QZ0p7EqvqxpClz_YH3XtlxgPmyXKcK-CGg_B_p5Z7w0,23
 nextrec/cli.py,sha256=uOaXnlAM-ARrbxKOVWWkTE_rv-54px168kBhFUHtIAg,25073
 nextrec/basic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nextrec/basic/activation.py,sha256=uekcJsOy8SiT0_NaDO2VNSStyYFzVikDFVLDk-VrjwQ,2949
@@ -14,10 +14,10 @@ nextrec/basic/model.py,sha256=4vBp-vXAWC5Oiu_x4mtVaXTKJCcKDYT0IJ7UOyHD5lw,110162
 nextrec/basic/session.py,sha256=mrIsjRJhmvcAfoO1pXX-KB3SK5CCgz89wH8XDoAiGEI,4475
 nextrec/basic/summary.py,sha256=b6jLo70gqZj_bQ4eb5yb8SXmr2ilZlKNN293EyVnkyc,17759
 nextrec/data/__init__.py,sha256=YZQjpty1pDCM7q_YNmiA2sa5kbujUw26ObLHWjMPjKY,1194
-nextrec/data/batch_utils.py,sha256=0bYGVX7RlhnHv_ZBaUngjDIpBNw-igCk98DgOsF7T6o,2879
+nextrec/data/batch_utils.py,sha256=vqlBvdfrhFEIUHmiWuUdXg1oY2jkyKOO1JO2l2KpJUU,3549
 nextrec/data/data_processing.py,sha256=lhuwYxWp4Ts2bbuLGDt2LmuPrOy7pNcKczd2uVcQ4ss,6476
 nextrec/data/data_utils.py,sha256=0Ls1cnG9lBz0ovtyedw5vwp7WegGK_iF-F8e_3DEddo,880
-nextrec/data/dataloader.py,sha256=gTs4YC5tHHwTq0A9481KYK1XyloeN2dMVOjPAFehF_E,19972
+nextrec/data/dataloader.py,sha256=2sXwoiWxupKE-V1QYeZlXjK1yJyxhDtlOhknAnJF8Wk,19727
 nextrec/data/preprocessor.py,sha256=AD5bHNbkAZAnI_SbDfJJaAh57CRtRjoOQJ6aIBkgoQs,65251
 nextrec/loss/__init__.py,sha256=rualGsY-IBvmM52q9eOBk0MyKcMkpkazcscOeDXi_SM,774
 nextrec/loss/grad_norm.py,sha256=YoE_XSIN1HOUcNq1dpfkIlWtMaB5Pu-SEWDaNgtRw1M,8316
@@ -36,7 +36,7 @@ nextrec/models/multi_task/escm.py,sha256=Djk4SjLsZ8He3sZrKfZkHdH2istR9R71GVEJc6E
 nextrec/models/multi_task/esmm.py,sha256=QRnNXV1IEArYorYJLOoBcLbxarMC_7azQLP79QTfAJ8,5273
 nextrec/models/multi_task/hmoe.py,sha256=6mTzZxC5PfSQovrmnR0O2hdhDUG_8yNqMwCdkNRlHkY,7567
 nextrec/models/multi_task/mmoe.py,sha256=Mplzstu-LYEMtiKiyiMEQZwY-xMkUeG-1FaDqqK5kws,7606
-nextrec/models/multi_task/pepnet.py,sha256=1da-uS5Dnl-lHs1SJrja-mQTlPJng3lKJcIRavWtFCk,13385
+nextrec/models/multi_task/pepnet.py,sha256=ikCDu66TTPdfp94HU-4QzKCW5JOEvwAf5E0QEQ0SZzw,13510
 nextrec/models/multi_task/ple.py,sha256=cO-NqEm-UZKRz2MznBjqsXL8ImH7WU1HRzXdWAtb7Kk,12089
 nextrec/models/multi_task/poso.py,sha256=Xjw9JBAiGR9CGewp1uS4b1soA7fOvSWTsIT6pS9_o30,18215
 nextrec/models/multi_task/share_bottom.py,sha256=BT-nu0NZTV4HlFkva_KnoKLSxB0-gYuJWPw7PRDGwC8,5172
@@ -83,8 +83,8 @@ nextrec/utils/loss.py,sha256=GBWQGpDaYkMJySpdG078XbeUNXUC34PVqFy0AqNS9N0,4578
 nextrec/utils/model.py,sha256=PI9y8oWz1lhktgapZsiXb8rTr2NrFFlc80tr4yOFHik,5334
 nextrec/utils/torch_utils.py,sha256=UQpWS7F3nITYqvx2KRBaQJc9oTowRkIvowhuQLt6NFM,11953
 nextrec/utils/types.py,sha256=VhtLXUVvu0zAZVAUgRUML4FExRC-GH-ZmC1UiVSr3HE,1523
-nextrec-0.4.27.dist-info/METADATA,sha256=oKaTQkJVnQjpSxH-rjWCFICRoIRGMbopjv6K3sXV_m8,23188
-nextrec-0.4.27.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-nextrec-0.4.27.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
-nextrec-0.4.27.dist-info/licenses/LICENSE,sha256=2fQfVKeafywkni7MYHyClC6RGGC3laLTXCNBx-ubtp0,1064
-nextrec-0.4.27.dist-info/RECORD,,
+nextrec-0.4.28.dist-info/METADATA,sha256=dZDTiMVnLG8aZga9anG-7iEZ8lGGceYOnoR3ZuTX-2I,23188
+nextrec-0.4.28.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+nextrec-0.4.28.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
+nextrec-0.4.28.dist-info/licenses/LICENSE,sha256=2fQfVKeafywkni7MYHyClC6RGGC3laLTXCNBx-ubtp0,1064
+nextrec-0.4.28.dist-info/RECORD,,