nextrec 0.4.11__py3-none-any.whl → 0.4.13__py3-none-any.whl

nextrec/models/ranking/ffm.py

@@ -0,0 +1,275 @@
+"""
+Date: create on 19/12/2025
+Checkpoint: edit on 19/12/2025
+Author: Yang Zhou, zyaztec@gmail.com
+Reference:
+[1] Juan Y, Zhuang Y, Chin W-S, et al. Field-aware Factorization Machines for CTR
+Prediction[C]//RecSys. 2016: 43-50.
+
+Field-aware Factorization Machines (FFM) extend FM by learning a distinct
+embedding of each feature for every target field. For a pair of fields (i, j),
+FFM uses v_{i,f_j} · v_{j,f_i} instead of a shared embedding, enabling richer
+context-aware interactions and stronger CTR performance on sparse categorical
+data.
+
+Pipeline:
+  (1) Build field-aware embeddings v_{i,f} for each feature i toward every field f
+  (2) Compute first-order linear terms for sparse/sequence (and optional dense) fields
+  (3) For each field pair (i, j), compute v_{i,f_j} · v_{j,f_i}
+  (4) Sum linear + field-aware interaction logits and output prediction
+
+Key Advantages:
+- Field-aware embeddings capture asymmetric interactions between fields
+- Improves CTR accuracy on sparse categorical features
+- Retains interpretable second-order structure
+
+FFM 在 FM 基础上引入字段感知机制：每个特征在不同目标字段下拥有不同的 embedding。
+对于字段对 (i, j)，模型使用 v_{i,f_j} 与 v_{j,f_i} 的内积，从而更细粒度地建模
+跨字段交互，在稀疏高维 CTR 场景中表现更优。
+
+处理流程：
+  (1) 为每个特征 i 构造面向每个字段 f 的嵌入 v_{i,f}
+  (2) 计算一阶线性项（稀疏/序列特征，及可选的稠密特征）
+  (3) 对每一对字段 (i, j) 计算 v_{i,f_j} · v_{j,f_i}
+  (4) 将线性项与交互项相加得到最终预测
+
+主要优点：
+- 字段感知嵌入可捕捉非对称交互
+- 稀疏类别特征下预测更准确
+- 保持二阶结构的可解释性
+"""
+
+import torch
+import torch.nn as nn
+
+from nextrec.basic.features import DenseFeature, SequenceFeature, SparseFeature
+from nextrec.basic.layers import AveragePooling, InputMask, PredictionLayer, SumPooling
+from nextrec.basic.model import BaseModel
+from nextrec.utils.torch_utils import get_initializer
+
+
+class FFM(BaseModel):
+    @property
+    def model_name(self):
+        return "FFM"
+
+    @property
+    def default_task(self):
+        return "binary"
+
+    def __init__(
+        self,
+        dense_features: list[DenseFeature] | None = None,
+        sparse_features: list[SparseFeature] | None = None,
+        sequence_features: list[SequenceFeature] | None = None,
+        target: list[str] | str | None = None,
+        task: str | list[str] | None = None,
+        optimizer: str = "adam",
+        optimizer_params: dict | None = None,
+        loss: str | nn.Module | None = "bce",
+        loss_params: dict | list[dict] | None = None,
+        device: str = "cpu",
+        embedding_l1_reg=1e-6,
+        dense_l1_reg=1e-5,
+        embedding_l2_reg=1e-5,
+        dense_l2_reg=1e-4,
+        **kwargs,
+    ):
+        dense_features = dense_features or []
+        sparse_features = sparse_features or []
+        sequence_features = sequence_features or []
+        optimizer_params = optimizer_params or {}
+        if loss is None:
+            loss = "bce"
+
+        super(FFM, self).__init__(
+            dense_features=dense_features,
+            sparse_features=sparse_features,
+            sequence_features=sequence_features,
+            target=target,
+            task=task or self.default_task,
+            device=device,
+            embedding_l1_reg=embedding_l1_reg,
+            dense_l1_reg=dense_l1_reg,
+            embedding_l2_reg=embedding_l2_reg,
+            dense_l2_reg=dense_l2_reg,
+            **kwargs,
+        )
+
+        self.fm_features = sparse_features + sequence_features
+        if len(self.fm_features) < 2:
+            raise ValueError(
+                "FFM requires at least two sparse/sequence features to build field-aware interactions."
+            )
+
+        self.embedding_dim = self.fm_features[0].embedding_dim
+        if any(f.embedding_dim != self.embedding_dim for f in self.fm_features):
+            raise ValueError(
+                "All FFM features must share the same embedding_dim for field-aware interactions."
+            )
+        for feature in self.fm_features:
+            if isinstance(feature, SequenceFeature) and feature.combiner == "concat":
+                raise ValueError(
+                    "FFM does not support SequenceFeature with combiner='concat' because it breaks shared embedding_dim."
+                )
+
+        self.field_aware_embeddings = nn.ModuleDict()
+        for src_feature in self.fm_features:
+            for target_field in self.fm_features:
+                key = self.field_aware_key(src_feature, target_field)
+                if key in self.field_aware_embeddings:
+                    continue
+                self.field_aware_embeddings[key] = self.build_embedding(src_feature)
+
+        # First-order terms for sparse/sequence features: one hot
+        self.first_order_embeddings = nn.ModuleDict()
+        for feature in self.fm_features:
+            if feature.embedding_name in self.first_order_embeddings:
+                continue
+            emb = nn.Embedding(
+                num_embeddings=feature.vocab_size,
+                embedding_dim=1,
+                padding_idx=feature.padding_idx,
+            )
+            self.first_order_embeddings[feature.embedding_name] = emb
+
+        # Optional dense linear term
+        self.dense_features = list(dense_features)
+        dense_input_dim = sum([f.input_dim for f in self.dense_features])
+        self.linear_dense = (
+            nn.Linear(dense_input_dim, 1, bias=True) if dense_input_dim > 0 else None
+        )
+
+        self.prediction_layer = PredictionLayer(task_type=self.task)
+        self.input_mask = InputMask()
+        self.mean_pool = AveragePooling()
+        self.sum_pool = SumPooling()
+
+        self.embedding_params.extend(
+            emb.weight for emb in self.field_aware_embeddings.values()
+        )
+        self.embedding_params.extend(
+            emb.weight for emb in self.first_order_embeddings.values()
+        )
+        self.register_regularization_weights(
+            embedding_attr="field_aware_embeddings", include_modules=["linear_dense"]
+        )
+
+        self.compile(
+            optimizer=optimizer,
+            optimizer_params=optimizer_params,
+            loss=loss,
+            loss_params=loss_params,
+        )
+
+    def field_aware_key(
+        self, src_feature: SparseFeature | SequenceFeature, target_field
+    ) -> str:
+        return f"{src_feature.embedding_name}__to__{target_field.name}"
+
+    def build_embedding(self, feature: SparseFeature | SequenceFeature) -> nn.Embedding:
+        if getattr(feature, "pretrained_weight", None) is not None:
+            weight = feature.pretrained_weight
+            if weight is None:
+                raise ValueError(
+                    f"[FFM Error]: Pretrained weight for '{feature.embedding_name}' is None."
+                )
+            if weight.shape != (feature.vocab_size, feature.embedding_dim):
+                raise ValueError(
+                    f"[FFM Error]: Pretrained weight for '{feature.embedding_name}' has shape {weight.shape}, expected ({feature.vocab_size}, {feature.embedding_dim})."
+                )
+            embedding = nn.Embedding.from_pretrained(
+                embeddings=weight,
+                freeze=feature.freeze_pretrained,
+                padding_idx=feature.padding_idx,
+            )
+            embedding.weight.requires_grad = (
+                feature.trainable and not feature.freeze_pretrained
+            )
+        else:
+            embedding = nn.Embedding(
+                num_embeddings=feature.vocab_size,
+                embedding_dim=feature.embedding_dim,
+                padding_idx=feature.padding_idx,
+            )
+            embedding.weight.requires_grad = feature.trainable
+            initialization = get_initializer(
+                init_type=feature.init_type,
+                activation="linear",
+                param=feature.init_params,
+            )
+            initialization(embedding.weight)
+        return embedding
+
+    def embed_for_field(
+        self,
+        feature: SparseFeature | SequenceFeature,
+        target_field,
+        x: dict[str, torch.Tensor],
+    ) -> torch.Tensor:
+        key = self.field_aware_key(feature, target_field)
+        emb = self.field_aware_embeddings[key]
+        if isinstance(feature, SparseFeature):
+            return emb(x[feature.name].long())
+
+        seq_input = x[feature.name].long()
+        if feature.max_len is not None and seq_input.size(1) > feature.max_len:
+            seq_input = seq_input[:, -feature.max_len :]
+        seq_emb = emb(seq_input)  # [B, L, D]
+        mask = self.input_mask(x, feature, seq_input)
+        if feature.combiner == "mean":
+            return self.mean_pool(seq_emb, mask)
+        if feature.combiner == "sum":
+            return self.sum_pool(seq_emb, mask)
+        raise ValueError(
+            f"[FFM Error]: Unsupported combiner '{feature.combiner}' for sequence feature '{feature.name}'."
+        )
+
+    def forward(self, x):
+        batch_size = x[self.fm_features[0].name].size(0)
+        device = x[self.fm_features[0].name].device
+        y_linear = torch.zeros(batch_size, 1, device=device)
+
+        # First-order dense part
+        if self.linear_dense is not None:
+            dense_inputs = [
+                x[f.name].float().view(batch_size, -1) for f in self.dense_features
+            ]
+            dense_stack = torch.cat(dense_inputs, dim=1) if dense_inputs else None
+            if dense_stack is not None:
+                y_linear = y_linear + self.linear_dense(dense_stack)
+
+        # First-order sparse/sequence part
+        first_order_terms = []
+        for feature in self.fm_features:
+            emb = self.first_order_embeddings[feature.embedding_name]
+            if isinstance(feature, SparseFeature):
+                term = emb(x[feature.name].long())  # [B, 1]
+            else:
+                seq_input = x[feature.name].long()
+                if feature.max_len is not None and seq_input.size(1) > feature.max_len:
+                    seq_input = seq_input[:, -feature.max_len :]
+                mask = self.input_mask(x, feature, seq_input).squeeze(1)  # [B, L]
+                seq_weight = emb(seq_input).squeeze(-1)  # [B, L]
+                term = (seq_weight * mask).sum(dim=1, keepdim=True)  # [B, 1]
+            first_order_terms.append(term)
+        if first_order_terms:
+            y_linear = y_linear + torch.sum(
+                torch.stack(first_order_terms, dim=1), dim=1
+            )
+
+        # Field-aware interactions
+        y_interaction = torch.zeros(batch_size, 1, device=device)
+        num_fields = len(self.fm_features)
+        for i in range(num_fields - 1):
+            feature_i = self.fm_features[i]
+            for j in range(i + 1, num_fields):
+                feature_j = self.fm_features[j]
+                v_i_fj = self.embed_for_field(feature_i, feature_j, x)
+                v_j_fi = self.embed_for_field(feature_j, feature_i, x)
+                y_interaction = y_interaction + torch.sum(
+                    v_i_fj * v_j_fi, dim=1, keepdim=True
+                )
+
+        y = y_linear + y_interaction
+        return self.prediction_layer(y)

nextrec/models/ranking/lr.py

@@ -113,8 +113,6 @@ class LR(BaseModel):
         )
 
     def forward(self, x):
-        input_linear = self.embedding(
-            x=x, features=self.all_features, squeeze_dim=True
-        )
+        input_linear = self.embedding(x=x, features=self.all_features, squeeze_dim=True)
         y = self.linear(input_linear)
         return self.prediction_layer(y)

nextrec/utils/__init__.py

@@ -36,7 +36,7 @@ from .data import (
 )
 from .embedding import get_auto_embedding_dim
 from .feature import normalize_to_list
-from .model import get_mlp_output_dim, merge_features
+from .model import compute_pair_scores, get_mlp_output_dim, merge_features
 from .torch_utils import (
     add_distributed_sampler,
     concat_tensors,
@@ -88,6 +88,7 @@ __all__ = [
     # Model utilities
     "merge_features",
     "get_mlp_output_dim",
+    "compute_pair_scores",
     # Feature utilities
     "normalize_to_list",
     # Config utilities

nextrec/utils/console.py

@@ -4,7 +4,7 @@ Console and CLI utilities for NextRec.
 This module centralizes CLI logging helpers, progress display, and metric tables.
 
 Date: create on 19/12/2025
-Checkpoint: edit on 19/12/2025
+Checkpoint: edit on 20/12/2025
 Author: Yang Zhou, zyaztec@gmail.com
 """
 
@@ -242,6 +242,14 @@ def display_metrics_table(
                 normalized_order.append(name)
         task_order = normalized_order
 
+    if not task_order and not grouped and not metrics:
+        if isinstance(loss, numbers.Number):
+            msg = f"Epoch {epoch}/{epochs} - {split} (loss={float(loss):.4f})"
+            if colorize is not None:
+                msg = colorize(msg)
+            logging.info(msg)
+        return
+
     if Console is None or Table is None or box is None:
         prefix = f"Epoch {epoch}/{epochs} - {split}:"
         segments: list[str] = []

nextrec/utils/model.py

@@ -7,6 +7,8 @@ Author: Yang Zhou, zyaztec@gmail.com
 
 from collections import OrderedDict
 
+import torch
+
 
 def merge_features(primary, secondary) -> list:
     merged: OrderedDict[str, object] = OrderedDict()
@@ -42,3 +44,15 @@ def select_features(
         )
 
     return [feature_map[name] for name in names]
+
+
+def compute_pair_scores(model, data, batch_size: int = 512):
+    user_emb = model.encode_user(data, batch_size=batch_size)
+    item_emb = model.encode_item(data, batch_size=batch_size)
+    with torch.no_grad():
+        user_tensor = torch.as_tensor(user_emb, device=model.device)
+        item_tensor = torch.as_tensor(item_emb, device=model.device)
+        scores = model.compute_similarity(user_tensor, item_tensor)
+        if model.training_mode == "pointwise":
+            scores = torch.sigmoid(scores)
+    return scores.detach().cpu().numpy()

{nextrec-0.4.11.dist-info → nextrec-0.4.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextrec
-Version: 0.4.11
+Version: 0.4.13
 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
@@ -66,7 +66,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.11-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.4.13-orange.svg)
 
 中文文档 | [English Version](README_en.md)
 
@@ -99,7 +99,7 @@ NextRec是一个基于PyTorch的现代推荐系统框架，旨在为研究工程
 
 ## NextRec近期进展
 
-- **12/12/2025** 在v0.4.11中加入了[RQ-VAE](/nextrec/models/representation/rqvae.py)模块。配套的[数据集](/dataset/ecommerce_task.csv)和[代码](tutorials/notebooks/zh/使用RQ-VAE构建语义ID.ipynb)已经同步在仓库中
+- **12/12/2025** 在v0.4.13中加入了[RQ-VAE](/nextrec/models/representation/rqvae.py)模块。配套的[数据集](/dataset/ecommerce_task.csv)和[代码](tutorials/notebooks/zh/使用RQ-VAE构建语义ID.ipynb)已经同步在仓库中
 - **07/12/2025** 发布了NextRec CLI命令行工具，它允许用户根据配置文件进行一键训练和推理，我们提供了相关的[教程](/nextrec_cli_preset/NextRec-CLI_zh.md)和[教学代码](/nextrec_cli_preset)
 - **03/12/2025** NextRec获得了100颗🌟！感谢大家的支持
 - **06/12/2025** 在v0.4.1中支持了单机多卡的分布式DDP训练，并且提供了配套的[代码](tutorials/distributed)
@@ -128,7 +128,7 @@ pip install nextrec # or pip install -e .
 - [movielen_ranking_deepfm.py](/tutorials/movielen_ranking_deepfm.py) - movielen 100k数据集上的 DeepFM 模型训练示例
 - [example_ranking_din.py](/tutorials/example_ranking_din.py) - 电商数据集上的DIN 深度兴趣网络训练示例
 - [example_multitask.py](/tutorials/example_multitask.py) - 电商数据集上的ESMM多任务学习训练示例
-- [movielen_match_dssm.py](/tutorials/example_match_dssm.py) - 基于movielen 100k数据集训练的 DSSM 召回模型示例
+- [movielen_match_dssm.py](/tutorials/movielen_match_dssm.py) - 基于movielen 100k数据集训练的 DSSM 召回模型示例
 
 - [example_distributed_training.py](/tutorials/distributed/example_distributed_training.py) - 使用NextRec进行单机多卡训练的代码示例
 
@@ -240,11 +240,11 @@ nextrec --mode=train --train_config=path/to/train_config.yaml
 nextrec --mode=predict --predict_config=path/to/predict_config.yaml
 ```
 
-> 截止当前版本0.4.11，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
+> 截止当前版本0.4.13，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
 
 ## 兼容平台
 
-当前最新版本为0.4.11，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
+当前最新版本为0.4.13，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
 
 | 平台 | 配置 | 
 |------|------|
@@ -262,7 +262,7 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
 | [FM](nextrec/models/ranking/fm.py) | Factorization Machines | ICDM 2010 | 已支持 |
 | [LR](nextrec/models/ranking/lr.py) | Logistic Regression | - | 已支持 |
 | [AFM](nextrec/models/ranking/afm.py) | Attentional Factorization Machines: Learning the Weight of Feature Interactions via Attention Networks | IJCAI 2017 | 已支持 |
-| [FFM](nextrec/models/ranking/ffm.py) | Field-aware Factorization Machines | RecSys 2010 | 开发中 |
+| [FFM](nextrec/models/ranking/ffm.py) | Field-aware Factorization Machines | RecSys 2016 | 已支持 |
 | [DeepFM](nextrec/models/ranking/deepfm.py) | DeepFM: A Factorization-Machine based Neural Network for CTR Prediction | IJCAI 2017 | 已支持 |
 | [Wide&Deep](nextrec/models/ranking/widedeep.py) | Wide & Deep Learning for Recommender Systems | DLRS 2016 | 已支持 |
 | [xDeepFM](nextrec/models/ranking/xdeepfm.py) | xDeepFM: Combining Explicit and Implicit Feature Interactions | KDD 2018 | 已支持 |

{nextrec-0.4.11.dist-info → nextrec-0.4.13.dist-info}/RECORD

@@ -1,34 +1,35 @@
 nextrec/__init__.py,sha256=_M3oUqyuvQ5k8Th_3wId6hQ_caclh7M5ad51XN09m98,235
-nextrec/__version__.py,sha256=xIphSmmFF5C8ZjsK5bpruTtbjrTL9bI6TjdjgsELGCw,23
-nextrec/cli.py,sha256=PXRNXMRm_a_1u6StnjsHefq0rKqsc6Mzx3mZmc9553g,23803
+nextrec/__version__.py,sha256=ARFl7G-gCe12exBb-FIsJnbsUD5V9okxkHUUdQqb0RA,23
+nextrec/cli.py,sha256=6nBY8O8-0931h428eQS8CALkKn1FmizovJme7Q1c_O0,23978
 nextrec/basic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nextrec/basic/activation.py,sha256=uzTWfCOtBSkbu_Gk9XBNTj8__s241CaYLJk6l8nGX9I,2885
-nextrec/basic/callback.py,sha256=a6gg7r3x1v0xaOSya9PteLql7I14nepY7gX8tDYtins,14679
-nextrec/basic/features.py,sha256=Wnbzr7UMotgv1Vzeg0o9Po-KKIvYUSYIghoVDfMPx_g,4340
-nextrec/basic/layers.py,sha256=GJH2Tx3IkZrYGb7-ET976iHCC28Ubck_NO9-iyY4mDI,28911
-nextrec/basic/loggers.py,sha256=JnQiFvmsVgZ63gqBLR2ZFWrVPzkxRbzWhTdeoiJKcos,6526
-nextrec/basic/metrics.py,sha256=8RswR_3MGvIBkT_n6fnmON2eYH-hfD7kIKVnyJJjL3o,23131
-nextrec/basic/model.py,sha256=OCcV9nTAZukurRISzPGCQM5yJ0Fpph3vOMKb2CPkI68,98685
+nextrec/basic/callback.py,sha256=nn1f8FG9c52vJ-gvwteqPbk3-1QuNS1vmhBlkENdb0I,14636
+nextrec/basic/features.py,sha256=GyCUzGPuizUofrZSSOdqHK84YhnX4MGTdu7Cx2OGhUA,4654
+nextrec/basic/layers.py,sha256=ZM3Nka3e2cit3e3peL0ukJCMgKZK1ovNFfAWvVOwlos,28556
+nextrec/basic/loggers.py,sha256=Zh1A5DVAFqlGglyaQ4_IMgvFbWAcXX5H3aHbCWA82nE,6524
+nextrec/basic/metrics.py,sha256=saNgM7kuHk9xqDxZF6x33irTaxeXCU-hxYTUQauuGgg,23074
+nextrec/basic/model.py,sha256=b_O81WSv1XxBAS5oQk92DlLdYAtnikr_epaV5T9RSxs,102570
 nextrec/basic/session.py,sha256=UOG_-EgCOxvqZwCkiEd8sgNV2G1sm_HbzKYVQw8yYDI,4483
 nextrec/data/__init__.py,sha256=YZQjpty1pDCM7q_YNmiA2sa5kbujUw26ObLHWjMPjKY,1194
 nextrec/data/batch_utils.py,sha256=0bYGVX7RlhnHv_ZBaUngjDIpBNw-igCk98DgOsF7T6o,2879
 nextrec/data/data_processing.py,sha256=lKXDBszrO5fJMAQetgSPr2mSQuzOluuz1eHV4jp0TDU,5538
 nextrec/data/data_utils.py,sha256=0Ls1cnG9lBz0ovtyedw5vwp7WegGK_iF-F8e_3DEddo,880
 nextrec/data/dataloader.py,sha256=xTORNbaQVa20sk2S3kyV0SSngscvq8bNqHr0AmYjFqM,18768
-nextrec/data/preprocessor.py,sha256=wNjivq2N-iDzBropkp3YfSkN0jSA4l4h81C-ECa6k4c,44643
-nextrec/loss/__init__.py,sha256=-sibZK8QXLblVNWqdqjrPPzMCDyIXSq7yd2eZ57p9Nw,810
+nextrec/data/preprocessor.py,sha256=K-cUP-YdlQx1VJ2m1CXuprncpjDJe2ERVO5xCSoxHKI,44470
+nextrec/loss/__init__.py,sha256=ZCgsfyR5YAecv6MdOsnUjkfacvZg2coQVjuKAfPvmRo,923
+nextrec/loss/grad_norm.py,sha256=91Grspx95Xu_639TkL_WZRX1xt5QOTZCzBeJWbUGPiE,8385
 nextrec/loss/listwise.py,sha256=UT9vJCOTOQLogVwaeTV7Z5uxIYnngGdxk-p9e97MGkU,5744
-nextrec/loss/loss_utils.py,sha256=Eg_EKm47onSCLhgs2q7IkB7TV9TwV1Dz4QgVR2yh-gc,4610
+nextrec/loss/loss_utils.py,sha256=xMmT_tWcKah_xcU3FzVMmSEzyZfxiMKZWUbwkAspcDg,4579
 nextrec/loss/pairwise.py,sha256=X9yg-8pcPt2IWU0AiUhWAt3_4W_3wIF0uSdDYTdoPFY,3398
 nextrec/loss/pointwise.py,sha256=o9J3OznY0hlbDsUXqn3k-BBzYiuUH5dopz8QBFqS_kQ,7343
 nextrec/models/generative/__init__.py,sha256=0MV3P-_ainPaTxmRBGWKUVCEt14KJvuvEHmRB3OQ1Fs,176
 nextrec/models/generative/tiger.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nextrec/models/multi_task/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/models/multi_task/esmm.py,sha256=3QQePhSkOcM4t472S4E5xU9_KiLiSwHb9CfdkEgmqqk,6491
-nextrec/models/multi_task/mmoe.py,sha256=uFTbc0MiFBDTCIt8mTW6xs0oyOn1EesIHHZo81HR35k,8583
-nextrec/models/multi_task/ple.py,sha256=z32etizNlTLwwR7CYKxy8u9owAbtiRh492Fje_y64hQ,13016
-nextrec/models/multi_task/poso.py,sha256=foH7XDUz0XN0s0zoyHLuTmrcs3QOT8-x4YGxLX1Lxxg,19016
-nextrec/models/multi_task/share_bottom.py,sha256=rmEnsX3LA3pNsLKfG1ir5WDLdkSY-imO_ASiclirJiA,6519
+nextrec/models/multi_task/esmm.py,sha256=AqesBZ4tOFNm7POCrHZ90h1zWWSViZAYfydUVOh2dEU,6545
+nextrec/models/multi_task/mmoe.py,sha256=aaQKcx4PL_mAanW3tkjAR886KmMCHTdBuu4p9EIKQJo,8657
+nextrec/models/multi_task/ple.py,sha256=fqkujPFGxxQOO_6nBZEz_UcxLEUoX_vCJsk0YOpxTg4,13084
+nextrec/models/multi_task/poso.py,sha256=J_Btxhm9JpFJMdQQHNNf9mMRHOgO7j1ts6VN5o4qJnk,19193
+nextrec/models/multi_task/share_bottom.py,sha256=DTWm6fpLCLiXimD-qk_0YIKT_9THMFDrnx4GDViXc_g,6583
 nextrec/models/ranking/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nextrec/models/ranking/afm.py,sha256=96jGUPL4yTWobMIVBjHpOxl9AtAzCAGR8yw7Sy2JmdQ,10125
 nextrec/models/ranking/autoint.py,sha256=S6Cxnp1q2OErSYqmIix5P-b4qLWR-0dY6TMStuU6WLg,8109
@@ -37,11 +38,11 @@ nextrec/models/ranking/dcn_v2.py,sha256=QnqQbJsrtQp4mtvnBXFUVefKyr4dw-gHNWrCbO26
 nextrec/models/ranking/deepfm.py,sha256=aXoK59e2KaaPe5vfyFW4YiHbX4E2iG3gxFCxmWo8RHk,5200
 nextrec/models/ranking/dien.py,sha256=c7Zs85vxhOgKHg5s0QcSLCn1xXCCSD177TMERgM_v8g,18958
 nextrec/models/ranking/din.py,sha256=gdUhuKiKXBNOALbK8fGhlbSeuDT8agcEdNSrC_wveHc,9422
-nextrec/models/ranking/eulernet.py,sha256=mZTrD8rKbGbWMEeWpTl8mVimytLFJTLM5-LS_I3U6cw,13115
-nextrec/models/ranking/ffm.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nextrec/models/ranking/eulernet.py,sha256=SQr7M_6GI1u09jpxzRasQLFAPLXcmqff69waER6fiD8,12201
+nextrec/models/ranking/ffm.py,sha256=9t685SViSU1J0ESz-lrYSXhf4d4BWLNYZXReeVEk3e8,11262
 nextrec/models/ranking/fibinet.py,sha256=_eroddVHooJcaGT8MqS4mUrtv5j4pnTmfI3FoAKOZhs,7919
 nextrec/models/ranking/fm.py,sha256=SsrSKK3y4xg5Lv-t3JLnZan55Hzze2AxAiVPuscy0bk,4536
-nextrec/models/ranking/lr.py,sha256=Qf8RozgWlsKjHGVbo-94d2Z_4kMfCXHmvwkYu3WVZjQ,4030
+nextrec/models/ranking/lr.py,sha256=MUonlKyA77_bfshTupFjOhY5tiuSJxApFM-_yOk4Nwk,4008
 nextrec/models/ranking/masknet.py,sha256=tY1y2lO0iq82oylPN0SBnL5Bikc8weinFXpURyVT1hE,12373
 nextrec/models/ranking/pnn.py,sha256=FcNIFAw5J0ORGSR6L8ZK7NeXlJPpojwe_SpsxMQqCFw,8174
 nextrec/models/ranking/widedeep.py,sha256=-ghKfe_0puvlI9fBQr8lK3gXkfVvslGwP40AJTGqc7w,5077
@@ -62,16 +63,16 @@ nextrec/models/retrieval/sdm.py,sha256=LhkCZSfGhxOxziEkUtjr_hnqcyciJ2qpMoBSFBVW9
 nextrec/models/retrieval/youtube_dnn.py,sha256=xtGPV6_5LeSZBKkrTaU1CmtxlhgYLvZmjpwYaXYIaEA,7403
 nextrec/models/sequential/hstu.py,sha256=P2Kl7HEL3afwiCApGKQ6UbUNO9eNXXrB10H7iiF8cI0,19735
 nextrec/models/sequential/sasrec.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/utils/__init__.py,sha256=5ss2XQq8QZ2Ko5eiQ7oIig5cIZNrYGIptaarYEeO7Fk,2550
+nextrec/utils/__init__.py,sha256=C-1l-suSsN_MlPlj_5LApyCRQLOao5l7bO0SccwKHw4,2598
 nextrec/utils/config.py,sha256=0HOeMyTlx8g6BZVpXzo2lEOkb-mzNwhbigQuUomsYnY,19934
-nextrec/utils/console.py,sha256=D2Vax9_b7bgvAAOyk-Q2oUhSk1B-OngY5buS9Gb9-I0,11398
+nextrec/utils/console.py,sha256=e94SiwA0gKn2pfpP94mY_jl-kFok3TCjxo298KdFuP4,11696
 nextrec/utils/data.py,sha256=alruiWZFbmwy3kO12q42VXmtHmXFFjVULpHa43fx_mI,21098
 nextrec/utils/embedding.py,sha256=akAEc062MG2cD7VIOllHaqtwzAirQR2gq5iW7oKpGAU,1449
 nextrec/utils/feature.py,sha256=rsUAv3ELyDpehVw8nPEEsLCCIjuKGTJJZuFaWB_wrPk,633
-nextrec/utils/model.py,sha256=dYl1XfIZt6aVjNyV2AAhcArwFRMcEAKrjG_pr8AVHs0,1163
+nextrec/utils/model.py,sha256=3B85a0IJCggI26dxv25IX8R_5yQPo7wXI0JIAns6bkQ,1727
 nextrec/utils/torch_utils.py,sha256=AKfYbSOJjEw874xsDB5IO3Ote4X7vnqzt_E0jJny0o8,13468
-nextrec-0.4.11.dist-info/METADATA,sha256=QlIiSVfAdB1MxGkkgshFLPkc1KbvhCTj2NfXe4lR71M,20957
-nextrec-0.4.11.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-nextrec-0.4.11.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
-nextrec-0.4.11.dist-info/licenses/LICENSE,sha256=2fQfVKeafywkni7MYHyClC6RGGC3laLTXCNBx-ubtp0,1064
-nextrec-0.4.11.dist-info/RECORD,,
+nextrec-0.4.13.dist-info/METADATA,sha256=BcBFpd0l4OdNRlXtG5R1UT-jMcAdloQJjOAG33E4KRE,20958
+nextrec-0.4.13.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+nextrec-0.4.13.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
+nextrec-0.4.13.dist-info/licenses/LICENSE,sha256=2fQfVKeafywkni7MYHyClC6RGGC3laLTXCNBx-ubtp0,1064
+nextrec-0.4.13.dist-info/RECORD,,