nextrec 0.3.6__tar.gz → 0.4.1__tar.gz

{nextrec-0.3.6 → nextrec-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextrec
-Version: 0.3.6
+Version: 0.4.1
 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
@@ -55,7 +55,7 @@ Requires-Dist: seaborn>=0.12.0; extra == 'dev'
 Description-Content-Type: text/markdown
 
 <p align="center">
-<img align="center" src="asserts/logo.png" width="40%">
+<img align="center" src="assets/logo.png" width="40%">
 <p>
 
 <div align="center">
@@ -63,7 +63,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.3.6-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.4.1-orange.svg)
 
 English | [中文文档](README_zh.md)
 
@@ -86,7 +86,7 @@ NextRec is a modern recommendation framework built on PyTorch, delivering a unif
 
 NextRec adopts a modular and low-coupling engineering design, enabling full-pipeline reusability and scalability across data processing → model construction → training & evaluation → inference & deployment. Its core components include: a Feature-Spec-driven Embedding architecture, the BaseModel abstraction, a set of independent reusable Layers, a unified DataLoader for both training and inference, and a ready-to-use Model Zoo.
 
-![NextRec Architecture](asserts/nextrec_diagram_en.png)
+![NextRec Architecture](assets/nextrec_diagram_en.png)
 
 > The project borrows ideas from excellent open-source rec libraries. Early layers referenced [torch-rechub](https://github.com/datawhalechina/torch-rechub) but have been replaced with in-house implementations. torch-rechub remains mature in architecture and models; the author contributed a bit there—feel free to check it out.
 
@@ -110,7 +110,7 @@ To dive deeper, Jupyter notebooks are available:
 - [Hands on the NextRec framework](/tutorials/notebooks/en/Hands%20on%20nextrec.ipynb)
 - [Using the data processor for preprocessing](/tutorials/notebooks/en/Hands%20on%20dataprocessor.ipynb)
 
-> Current version [0.3.6]: the matching module is not fully polished yet and may have compatibility issues or unexpected errors. Please raise an issue if you run into problems.
+> Current version [0.4.1]: the matching module is not fully polished yet and may have compatibility issues or unexpected errors. Please raise an issue if you run into problems.
 
 ## 5-Minute Quick Start
 
@@ -196,6 +196,16 @@ metrics = model.evaluate(
 )
 ```
 
+## Platform Compatibility
+
+The current version is 0.4.1. 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 | 
+|----------|---------------|
+| MacOS latest | MacBook Pro M4 Pro 24GB RAM |
+| Ubuntu latest | AutoDL 4070D Dual GPU |
+| CentOS 7 | Intel Xeon 5138Y 96 cores 377GB RAM |
+
 ---
 
 ## Supported Models

{nextrec-0.3.6 → nextrec-0.4.1}/README.md

@@ -1,5 +1,5 @@
 <p align="center">
-<img align="center" src="asserts/logo.png" width="40%">
+<img align="center" src="assets/logo.png" width="40%">
 <p>
 
 <div align="center">
@@ -7,7 +7,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.3.6-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.4.1-orange.svg)
 
 English | [中文文档](README_zh.md)
 
@@ -30,7 +30,7 @@ NextRec is a modern recommendation framework built on PyTorch, delivering a unif
 
 NextRec adopts a modular and low-coupling engineering design, enabling full-pipeline reusability and scalability across data processing → model construction → training & evaluation → inference & deployment. Its core components include: a Feature-Spec-driven Embedding architecture, the BaseModel abstraction, a set of independent reusable Layers, a unified DataLoader for both training and inference, and a ready-to-use Model Zoo.
 
-![NextRec Architecture](asserts/nextrec_diagram_en.png)
+![NextRec Architecture](assets/nextrec_diagram_en.png)
 
 > The project borrows ideas from excellent open-source rec libraries. Early layers referenced [torch-rechub](https://github.com/datawhalechina/torch-rechub) but have been replaced with in-house implementations. torch-rechub remains mature in architecture and models; the author contributed a bit there—feel free to check it out.
 
@@ -54,7 +54,7 @@ To dive deeper, Jupyter notebooks are available:
 - [Hands on the NextRec framework](/tutorials/notebooks/en/Hands%20on%20nextrec.ipynb)
 - [Using the data processor for preprocessing](/tutorials/notebooks/en/Hands%20on%20dataprocessor.ipynb)
 
-> Current version [0.3.6]: the matching module is not fully polished yet and may have compatibility issues or unexpected errors. Please raise an issue if you run into problems.
+> Current version [0.4.1]: the matching module is not fully polished yet and may have compatibility issues or unexpected errors. Please raise an issue if you run into problems.
 
 ## 5-Minute Quick Start
 
@@ -140,6 +140,16 @@ metrics = model.evaluate(
 )
 ```
 
+## Platform Compatibility
+
+The current version is 0.4.1. 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 | 
+|----------|---------------|
+| MacOS latest | MacBook Pro M4 Pro 24GB RAM |
+| Ubuntu latest | AutoDL 4070D Dual GPU |
+| CentOS 7 | Intel Xeon 5138Y 96 cores 377GB RAM |
+
 ---
 
 ## Supported Models

{nextrec-0.3.6 → nextrec-0.4.1}/README_zh.md

@@ -1,5 +1,5 @@
 <p align="center">
-<img align="center" src="asserts/logo.png" width="40%">
+<img align="center" src="assets/logo.png" width="40%">
 <p>
 
 <div align="center">
@@ -7,7 +7,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.3.6-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.4.1-orange.svg)
 
 [English Version](README.md) | 中文文档
 
@@ -17,7 +17,7 @@
 
 ## 简介
 
-NextRec 是一个基于 PyTorch 构建的现代推荐系统框架，为研究人员与工程团队提供统一的建模、训练与评估体验。框架采用模块化设计，内置丰富的模型实现、数据处理工具和工程化训练组件，可快速覆盖多种推荐场景，主要面向Spark集群下，基于大数据量parquet离线特征训练的工业推荐召回算法场景。
+NextRec 是一个基于 PyTorch 构建的现代推荐系统框架,为研究人员与工程团队提供统一的建模、训练与评估体验。框架采用模块化设计,内置丰富的模型实现、数据处理工具和工程化训练组件,可快速覆盖多种推荐场景,主要面向Spark集群下,基于大数据量parquet离线特征训练的工业推荐召回算法场景。
 
 ## Why NextRec
 
@@ -30,7 +30,7 @@ NextRec 是一个基于 PyTorch 构建的现代推荐系统框架，为研究人
 
 NextRec采用模块化、低耦合的工程设计，使得推荐系统从数据处理 → 模型构建 → 训练评估 → 推理部署 全链路都具备复用性与可扩展性。其中的核心组件包括：Feature Spec驱动的Embedding架构；模型基类BaseModel；独立Layer模块；支持训练和推理的统一的DataLoader；开箱即用的模型库。
 
-![NextRec架构](asserts/nextrec_diagram_zh.png)
+![NextRec架构](assets/nextrec_diagram_zh.png)
 
 > 项目的架构借鉴了一些优秀的开源推荐算法库，其中最初版的layer层借鉴了DataWhaleChina社区的[torch-rechub](https://github.com/datawhalechina/torch-rechub)，现已替换为了自有实现。torch-rechub在开发架构和模型实现上相对成熟，本人也参与了其中一小部分的维护，欢迎感兴趣的开发者前往了解。
 
@@ -54,7 +54,7 @@ NextRec采用模块化、低耦合的工程设计，使得推荐系统从数据
 - [如何上手NextRec框架](/tutorials/notebooks/zh/Hands%20on%20nextrec.ipynb)
 - [如何使用数据处理器进行数据预处理](/tutorials/notebooks/zh/Hands%20on%20dataprocessor.ipynb)
 
-> 当前版本[0.3.6]，召回模型模块尚不完善，可能存在一些兼容性问题或意外报错，如果遇到问题，欢迎开发者在Issue区提出问题。
+> 当前版本[0.4.1]，召回模型模块尚不完善，可能存在一些兼容性问题或意外报错，如果遇到问题，欢迎开发者在Issue区提出问题。
 
 ## 5分钟快速上手
 
@@ -138,6 +138,15 @@ metrics = model.evaluate(
     user_id_column='user_id'
 )
 ```
+## 兼容平台
+
+当前最新版本为0.4.1，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
+
+| 平台 | 配置 | 
+|------|------|
+| MacOS latest| MacBook Pro M4 Pro 24G内存 |
+| Ubuntu latest| AutoDL 4070D 双卡 |
+| CentOS 7 | Intel Xeon 5138Y 96核 377G内存 | 
 
 ## 支持模型
 

{nextrec-0.3.6 → nextrec-0.4.1}/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.3.6"
+release = "0.4.1"
 
 extensions = [
     "myst_parser",

nextrec-0.4.1/nextrec/__version__.py

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

{nextrec-0.3.6 → nextrec-0.4.1}/nextrec/basic/layers.py

@@ -80,9 +80,7 @@ class PredictionLayer(nn.Module):
             else:
                 raise ValueError(f"[PredictionLayer Error]: Unsupported task_type '{task_type}'.")
             outputs.append(activation(task_logits))
-        result = torch.cat(outputs, dim=-1)
-        if result.shape[-1] == 1:
-            result = result.squeeze(-1)
+        result = torch.cat(outputs, dim=-1)  # single: (N,1), multi-task/multi-class: (N,total_dim)
         return result
 
 class EmbeddingLayer(nn.Module):
@@ -235,14 +233,28 @@ class InputMask(nn.Module):
         super().__init__()
 
     def forward(self, x: dict[str, torch.Tensor], feature: SequenceFeature, seq_tensor: torch.Tensor | None = None):
-        values = seq_tensor if seq_tensor is not None else x[feature.name]
-        if feature.padding_idx is not None:
-            mask = (values.long() != feature.padding_idx)
+        if seq_tensor is not None:
+            values = seq_tensor
         else:
-            mask = (values.long() != 0)
+            values = x[feature.name]
+        values = values.long()
+        padding_idx = feature.padding_idx if feature.padding_idx is not None else 0
+        mask = (values != padding_idx) 
+
         if mask.dim() == 1:
-            mask = mask.unsqueeze(-1)
-        return mask.unsqueeze(1).float()
+            # [B] -> [B, 1, 1]
+            mask = mask.unsqueeze(1).unsqueeze(2)
+        elif mask.dim() == 2:
+            # [B, L] -> [B, 1, L]
+            mask = mask.unsqueeze(1)
+        elif mask.dim() == 3:
+            # [B, 1, L]
+            # [B, L, 1]  -> [B, L] -> [B, 1, L]
+            if mask.size(1) != 1 and mask.size(2) == 1:
+                mask = mask.squeeze(-1).unsqueeze(1)
+        else:
+            raise ValueError(f"InputMask only supports 1D/2D/3D tensors, got shape {values.shape}")
+        return mask.float()
 
 class LR(nn.Module):
     def __init__(
@@ -299,20 +311,25 @@ class MLP(nn.Module):
         super().__init__()
         if dims is None:
             dims = []
-        layers = list()
+        layers = []
+        current_dim = input_dim
+
         for i_dim in dims:
-            layers.append(nn.Linear(input_dim, i_dim))
+            layers.append(nn.Linear(current_dim, i_dim))
             layers.append(nn.BatchNorm1d(i_dim))
             layers.append(activation_layer(activation))
             layers.append(nn.Dropout(p=dropout))
-            input_dim = i_dim
+            current_dim = i_dim
+            
         if output_layer:
-            layers.append(nn.Linear(input_dim, 1))
+            layers.append(nn.Linear(current_dim, 1))
+            self.output_dim = 1
+        else:
+            self.output_dim = current_dim  
         self.mlp = nn.Sequential(*layers)
-
     def forward(self, x):
         return self.mlp(x)
-
+    
 class FM(nn.Module):
     def __init__(self, reduce_sum: bool = True):
         super().__init__()