nextrec 0.4.34__py3-none-any.whl → 0.5.1__py3-none-any.whl

nextrec/models/ranking/afm.py

@@ -156,7 +156,7 @@ class AFM(BaseModel):
         # 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
+                x[f.name].float().reshape(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:
@@ -170,7 +170,7 @@ class AFM(BaseModel):
                 term = emb(x[feature.name].long())  # [B, 1]
             else:  # SequenceFeature
                 seq_input = x[feature.name].long()  # [B, 1]
-                if feature.max_len is not None and seq_input.size(1) > feature.max_len:
+                if feature.max_len is not None:
                     seq_input = seq_input[:, -feature.max_len :]
                 mask = self.input_mask(x, feature, seq_input).squeeze(1)  # [B, 1]
                 seq_weight = emb(seq_input).squeeze(-1)  # [B, L]
@@ -186,16 +186,11 @@ class AFM(BaseModel):
         for feature in self.fm_features:
             value = x.get(f"{feature.name}_value")
             if value is not None:
-                value = value.float()
-                if value.dim() == 1:
-                    value = value.unsqueeze(-1)
+                value = value.float().reshape(batch_size, -1)
             else:
                 if isinstance(feature, SequenceFeature):
                     seq_input = x[feature.name].long()
-                    if (
-                        feature.max_len is not None
-                        and seq_input.size(1) > feature.max_len
-                    ):
+                    if feature.max_len is not None:
                         seq_input = seq_input[:, -feature.max_len :]
                     value = self.input_mask(x, feature, seq_input).sum(dim=2)  # [B, 1]
                 else:

nextrec/models/ranking/dien.py

@@ -390,13 +390,13 @@ class DIEN(BaseModel):
                 dim=-1,
             )
             score_t = self.attention_layer.attention_net(concat_feat)  # [B, 1]
-            att_scores_list.append(score_t)
+            att_scores_list.append(score_t.unsqueeze(1))
 
         # [B, seq_len, 1]
         att_scores = torch.cat(att_scores_list, dim=1)
 
-        scores_flat = att_scores.squeeze(-1)  # [B, seq_len]
-        mask_flat = mask.squeeze(-1)  # [B, seq_len]
+        scores_flat = att_scores[..., 0]  # [B, seq_len]
+        mask_flat = mask[..., 0]  # [B, seq_len]
 
         scores_flat = scores_flat.masked_fill(mask_flat == 0, -1e9)
         att_weights = torch.softmax(scores_flat, dim=1)  # [B, seq_len]
@@ -437,8 +437,7 @@ class DIEN(BaseModel):
 
         for feat in self.dense_features:
             val = x[feat.name].float()
-            if val.dim() == 1:
-                val = val.unsqueeze(1)
+            val = val.view(val.size(0), -1)
             other_embeddings.append(val)
 
         concat_input = torch.cat(other_embeddings, dim=-1)  # [B, total_dim]
@@ -460,15 +459,15 @@ class DIEN(BaseModel):
         interest_states = interest_states[:, :-1, :]
         pos_seq = behavior_emb[:, 1:, :]
         neg_seq = neg_behavior_emb[:, 1:, :]
-        aux_mask = mask[:, 1:, :].squeeze(-1)
+        aux_mask = mask[:, 1:, 0]
 
         if aux_mask.sum() == 0:
             return torch.tensor(0.0, device=self.device)
 
         pos_input = torch.cat([interest_states, pos_seq], dim=-1)
         neg_input = torch.cat([interest_states, neg_seq], dim=-1)
-        pos_logits = self.auxiliary_net(pos_input).squeeze(-1)
-        neg_logits = self.auxiliary_net(neg_input).squeeze(-1)
+        pos_logits = self.auxiliary_net(pos_input)[..., 0]
+        neg_logits = self.auxiliary_net(neg_input)[..., 0]
 
         pos_loss = F.binary_cross_entropy_with_logits(
             pos_logits, torch.ones_like(pos_logits), reduction="none"

nextrec/models/ranking/ffm.py

@@ -190,7 +190,7 @@ class FFM(BaseModel):
             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:
+        if feature.max_len is not None:
             seq_input = seq_input[:, -feature.max_len :]
         seq_emb = emb(seq_input)  # [B, L, D]
         mask = self.input_mask(x, feature, seq_input)
@@ -224,7 +224,7 @@ class FFM(BaseModel):
                 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:
+                if feature.max_len is not None:
                     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]

nextrec/models/retrieval/sdm.py

@@ -223,8 +223,7 @@ class SDM(BaseMatchModel):
             for feat in self.user_dense_features:
                 if feat.name in user_input:
                     val = user_input[feat.name].float()
-                    if val.dim() == 1:
-                        val = val.unsqueeze(1)
+                    val = val.reshape(val.size(0), -1)
                     dense_features.append(val)
             if dense_features:
                 features_list.append(torch.cat(dense_features, dim=1))

nextrec/models/sequential/hstu.py

@@ -438,8 +438,6 @@ class HSTU(BaseModel):
         return self.causal_mask[:seq_len, :seq_len]
 
     def trim_sequence(self, seq: torch.Tensor) -> torch.Tensor:
-        if seq.size(1) <= self.max_seq_len:
-            return seq
         return seq[:, -self.max_seq_len :]
 
     def forward(self, x: dict[str, torch.Tensor]) -> torch.Tensor:

nextrec/utils/onnx_utils.py

@@ -0,0 +1,252 @@
+"""
+ONNX utilities for NextRec.
+
+Date: create on 25/01/2026
+Author: Yang Zhou, zyaztec@gmail.com
+"""
+
+from __future__ import annotations
+
+from pathlib import Path
+from typing import Iterable, Sequence
+
+import torch
+import numpy as np
+import onnxruntime as ort
+
+from nextrec.basic.features import DenseFeature, SequenceFeature, SparseFeature
+from nextrec.utils.torch_utils import to_numpy
+
+
+class OnnxModelWrapper(torch.nn.Module):
+    """Wrap a NextRec model to accept positional ONNX inputs."""
+
+    def __init__(self, model: torch.nn.Module, feature_names: Sequence[str]):
+        super().__init__()
+        self.model = model
+        self.feature_names = list(feature_names)
+
+    def forward(self, *inputs: torch.Tensor):
+        if len(inputs) != len(self.feature_names):
+            raise ValueError(
+                "[OnnxWrapper Error] Number of inputs does not match feature names."
+            )
+        x = {name: tensor for name, tensor in zip(self.feature_names, inputs)}
+        output = self.model(x)
+        if isinstance(output, list):
+            return tuple(output)
+        return output
+
+
+def create_dummy_inputs(
+    features: Iterable[object],
+    batch_size: int,
+    device: torch.device,
+    default_seq_len: int = 10,
+    seq_len_map: dict[str, int] | None = None,
+) -> list[torch.Tensor]:
+    tensors: list[torch.Tensor] = []
+    for feature in features:
+        if isinstance(feature, DenseFeature):
+            input_dim = max(int(feature.input_dim), 1)
+            tensors.append(
+                torch.zeros((batch_size, input_dim), dtype=torch.float32, device=device)
+            )
+        elif isinstance(feature, SequenceFeature):
+            seq_len = None
+            if seq_len_map:
+                seq_len = seq_len_map.get(feature.name)
+            if seq_len is None:
+                seq_len = (
+                    feature.max_len if feature.max_len is not None else default_seq_len
+                )
+            seq_len = max(int(seq_len), 1)
+            tensors.append(
+                torch.zeros((batch_size, seq_len), dtype=torch.long, device=device)
+            )
+        else:
+            tensors.append(torch.zeros((batch_size,), dtype=torch.long, device=device))
+    return tensors
+
+
+def normalize_dense(feature: DenseFeature, array: object) -> np.ndarray:
+    arr = np.asarray(array, dtype=np.float32)
+    if arr.ndim == 1:
+        arr = arr.reshape(-1, 1)
+    else:
+        arr = arr.reshape(arr.shape[0], -1)
+    expected = max(int(feature.input_dim), 1)
+    if arr.shape[1] != expected:
+        raise ValueError(
+            f"[ONNX Input Error] Dense feature '{feature.name}' expects {expected} dims but got {arr.shape[1]}."
+        )
+    return arr
+
+
+def normalize_sparse(feature: SparseFeature, array: object) -> np.ndarray:
+    arr = np.asarray(array, dtype=np.int64)
+    if arr.ndim == 2 and arr.shape[1] == 1:
+        arr = arr.reshape(-1)
+    elif arr.ndim != 1:
+        arr = arr.reshape(arr.shape[0], -1)
+        if arr.shape[1] != 1:
+            raise ValueError(
+                f"[ONNX Input Error] Sparse feature '{feature.name}' expects 1 dim but got {arr.shape}."
+            )
+        arr = arr.reshape(-1)
+    return arr
+
+
+def normalize_sequence(feature: SequenceFeature, array: object) -> np.ndarray:
+    arr = np.asarray(array, dtype=np.int64)
+    if arr.ndim == 1:
+        arr = arr.reshape(1, -1)
+    elif arr.ndim > 2:
+        arr = arr.reshape(arr.shape[0], -1)
+    max_len = feature.max_len if feature.max_len is not None else arr.shape[1]
+    max_len = max(int(max_len), 1)
+    if arr.shape[1] > max_len:
+        arr = arr[:, :max_len]
+    elif arr.shape[1] < max_len:
+        pad_value = feature.padding_idx if feature.padding_idx is not None else 0
+        pad_width = max_len - arr.shape[1]
+        arr = np.pad(arr, ((0, 0), (0, pad_width)), constant_values=pad_value)
+    return arr
+
+
+def build_onnx_input_feed(
+    features: Iterable[object],
+    feature_batch: dict[str, object],
+    input_names: Sequence[str] | None = None,
+) -> dict[str, np.ndarray]:
+    feed: dict[str, np.ndarray] = {}
+    for feature in features:
+        if input_names is not None and feature.name not in input_names:
+            continue
+        if feature.name not in feature_batch:
+            raise KeyError(
+                f"[ONNX Input Error] Feature '{feature.name}' missing from batch data."
+            )
+        value = to_numpy(feature_batch[feature.name])
+        if isinstance(feature, DenseFeature):
+            value = normalize_dense(feature, value)
+        elif isinstance(feature, SequenceFeature):
+            value = normalize_sequence(feature, value)
+        else:
+            value = normalize_sparse(feature, value)
+        feed[feature.name] = value
+    return feed
+
+
+def pad_tensor(
+    value: torch.Tensor, pad_rows: int, pad_value: int | float
+) -> torch.Tensor:
+    if pad_rows <= 0:
+        return value
+    pad_shape = (pad_rows, *value.shape[1:])
+    pad = torch.full(pad_shape, pad_value, dtype=value.dtype, device=value.device)
+    return torch.cat([value, pad], dim=0)
+
+
+def pad_array(value: np.ndarray, pad_rows: int, pad_value: int | float) -> np.ndarray:
+    if pad_rows <= 0:
+        return value
+    pad_shape = (pad_rows, *value.shape[1:])
+    pad = np.full(pad_shape, pad_value, dtype=value.dtype)
+    return np.concatenate([value, pad], axis=0)
+
+
+def pad_onnx_inputs(
+    features: Iterable[object],
+    feature_batch: dict[str, object],
+    target_batch: int,
+) -> tuple[dict[str, object], int]:
+    if target_batch <= 0:
+        return feature_batch, 0
+    padded: dict[str, object] = {}
+    orig_batch = None
+    for feature in features:
+        if feature.name not in feature_batch:
+            continue
+        value = feature_batch[feature.name]
+        if isinstance(value, torch.Tensor):
+            batch = value.shape[0] if value.dim() > 0 else 1
+        else:
+            arr = np.asarray(value)
+            batch = arr.shape[0] if arr.ndim > 0 else 1
+        if orig_batch is None:
+            orig_batch = int(batch)
+        pad_rows = max(int(target_batch) - int(batch), 0)
+        if isinstance(feature, DenseFeature):
+            pad_value: int | float = 0.0
+        elif isinstance(feature, SequenceFeature):
+            pad_value = feature.padding_idx if feature.padding_idx is not None else 0
+        else:
+            pad_value = 0
+        if isinstance(value, torch.Tensor):
+            padded[feature.name] = pad_tensor(value, pad_rows, pad_value)
+        else:
+            padded[feature.name] = pad_array(np.asarray(value), pad_rows, pad_value)
+    if orig_batch is None:
+        orig_batch = 0
+    return padded, orig_batch
+
+
+def pad_id_batch(
+    id_batch: dict[str, object],
+    target_batch: int,
+    pad_value: int | float = 0,
+) -> tuple[dict[str, object], int]:
+    if target_batch <= 0:
+        return id_batch, 0
+    padded: dict[str, object] = {}
+    orig_batch = None
+    for name, value in id_batch.items():
+        if isinstance(value, torch.Tensor):
+            batch = value.shape[0] if value.dim() > 0 else 1
+        else:
+            arr = np.asarray(value)
+            batch = arr.shape[0] if arr.ndim > 0 else 1
+        if orig_batch is None:
+            orig_batch = int(batch)
+        pad_rows = max(int(target_batch) - int(batch), 0)
+        if isinstance(value, torch.Tensor):
+            padded[name] = pad_tensor(value, pad_rows, pad_value)
+        else:
+            padded[name] = pad_array(np.asarray(value), pad_rows, pad_value)
+    if orig_batch is None:
+        orig_batch = 0
+    return padded, orig_batch
+
+
+def merge_onnx_outputs(outputs: Sequence[np.ndarray]) -> np.ndarray:
+    if not outputs:
+        raise ValueError("[ONNX Output Error] Empty ONNX output list.")
+    if len(outputs) == 1:
+        return outputs[0]
+    normalized: list[np.ndarray] = []
+    batch = outputs[0].shape[0] if outputs[0].ndim > 0 else None
+    for out in outputs:
+        arr = np.asarray(out)
+        if arr.ndim == 0:
+            arr = arr.reshape(1, 1)
+        elif arr.ndim == 1:
+            arr = arr.reshape(-1, 1)
+        if batch is not None and arr.shape[0] != batch:
+            raise ValueError(
+                "[ONNX Output Error] Output batch size mismatch across ONNX outputs."
+            )
+        normalized.append(arr)
+    return np.concatenate(normalized, axis=1)
+
+
+def load_onnx_session(
+    onnx_path: str | Path,
+):
+    available = ort.get_available_providers()
+    preferred = ["CUDAExecutionProvider", "CPUExecutionProvider"]
+    selected = [p for p in preferred if p in available]
+    if not selected:
+        selected = available
+
+    return ort.InferenceSession(str(onnx_path), providers=selected)

nextrec/utils/torch_utils.py

@@ -61,7 +61,12 @@ def to_tensor(
 ) -> torch.Tensor:
     if value is None:
         raise ValueError("[Tensor Utils Error] Cannot convert None to tensor.")
-    tensor = value if isinstance(value, torch.Tensor) else torch.as_tensor(value)
+    if isinstance(value, torch.Tensor):
+        tensor = value
+    else:
+        if isinstance(value, np.ndarray):
+            value = value.copy()
+        tensor = torch.as_tensor(value)
     if tensor.dtype != dtype:
         tensor = tensor.to(dtype=dtype)
 

{nextrec-0.4.34.dist-info → nextrec-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextrec
-Version: 0.4.34
+Version: 0.5.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
@@ -24,10 +24,14 @@ Requires-Dist: numpy<2.0,>=1.21; sys_platform == 'linux' and python_version < '3
 Requires-Dist: numpy<3.0,>=1.26; sys_platform == 'linux' and python_version >= '3.12'
 Requires-Dist: numpy>=1.23.0; sys_platform == 'win32'
 Requires-Dist: numpy>=1.24.0; sys_platform == 'darwin'
+Requires-Dist: onnx>=1.16.0
+Requires-Dist: onnxruntime>=1.18.0
+Requires-Dist: onnxscript>=0.1.1
 Requires-Dist: pandas<2.0,>=1.5; sys_platform == 'linux' and python_version < '3.12'
 Requires-Dist: pandas<2.3.0,>=2.1.0; sys_platform == 'win32'
 Requires-Dist: pandas>=2.0.0; sys_platform == 'darwin'
 Requires-Dist: pandas>=2.1.0; sys_platform == 'linux' and python_version >= '3.12'
+Requires-Dist: polars>=0.20.0
 Requires-Dist: pyarrow<13.0.0,>=10.0.0; sys_platform == 'linux' and python_version < '3.12'
 Requires-Dist: pyarrow<15.0.0,>=12.0.0; sys_platform == 'win32'
 Requires-Dist: pyarrow>=12.0.0; sys_platform == 'darwin'
@@ -69,7 +73,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.34-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.5.1-orange.svg)
 [![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
 
 中文文档 | [English Version](README_en.md)
@@ -102,6 +106,7 @@ NextRec是一个基于PyTorch的现代推荐系统框架，旨在为研究工程
 - **高效训练与评估**：内置多种优化器、学习率调度、早停、模型检查点与详细的日志管理，开箱即用。
 
 ## NextRec近期进展
+- **28/01/2026** 在v0.4.39中加入了对onnx导出和加载的支持，并大大加速了数据预处理速度（最高9x加速）
 - **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)
 - **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'`进行配置
@@ -136,6 +141,7 @@ pip install nextrec # or pip install -e .
 - [example_multitask.py](/tutorials/example_multitask.py) - 电商数据集上的ESMM多任务学习训练示例
 - [movielen_match_dssm.py](/tutorials/movielen_match_dssm.py) - 基于movielen 100k数据集训练的 DSSM 召回模型示例
 
+- [example_onnx.py](/tutorials/example_onnx.py) - 使用NextRec训练和导出onnx模型
 - [example_distributed_training.py](/tutorials/distributed/example_distributed_training.py) - 使用NextRec进行单机多卡训练的代码示例
 
 - [run_all_ranking_models.py](/tutorials/run_all_ranking_models.py) - 快速校验所有排序模型的可用性
@@ -254,11 +260,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
 
 预测结果固定保存到 `{checkpoint_path}/predictions/{name}.{save_data_format}`。
 
-> 截止当前版本0.4.34，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
+> 截止当前版本0.5.1，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
 
 ## 兼容平台
 
-当前最新版本为0.4.34，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
+当前最新版本为0.5.1，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
 
 | 平台 | 配置 | 
 |------|------|

{nextrec-0.4.34.dist-info → nextrec-0.5.1.dist-info}/RECORD

@@ -1,24 +1,24 @@
 nextrec/__init__.py,sha256=_M3oUqyuvQ5k8Th_3wId6hQ_caclh7M5ad51XN09m98,235
-nextrec/__version__.py,sha256=lEcP5iYKQN3Lr-tUbNBN0qjKzIg4RjGwnIQfJt42ids,23
-nextrec/cli.py,sha256=ct6tnDRK3_M_dIjCt0W_KpEJdNgXBXY68SvGHsAgwU4,25874
+nextrec/__version__.py,sha256=eZ1bOun1DDVV0YLOBW4wj2FP1ajReLjbIrGmzN7ASBw,22
+nextrec/cli.py,sha256=lISpZYbdyQpKl6rWqm4ENXh4WqSfu-8jlC8jTDt1u1Y,29074
 nextrec/basic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/basic/activation.py,sha256=2A4sgkXzPUsGzBqu9kfGEw3IY2Z3MB22gXh12dAR-U8,2896
+nextrec/basic/activation.py,sha256=rU-W-DHgiD3AZnMGmD014ChxklfP9BpedDTiwtdgXhA,2762
 nextrec/basic/asserts.py,sha256=eaB4FZJ7Sbh9S8PLJZNiYd7Zi98ca2rDi4S7wSYCaEw,1473
 nextrec/basic/callback.py,sha256=H1C7EdkLTRLtPrKwCk1Gwq41m7I6tmtUWEST5Ih9fHI,12648
 nextrec/basic/features.py,sha256=hVbEEtEYFer5OLkqNEc0N0vK9QkkWE9Il8FOW0hejZQ,8528
 nextrec/basic/heads.py,sha256=WqvavaH6Y8Au8dLaoUfH2AaOOWgYvjZI5US8avkQNsQ,4009
-nextrec/basic/layers.py,sha256=ta2tLCERCoWXRzADb5ziGkxA2iftdiSB1bYJhdy55WQ,39526
+nextrec/basic/layers.py,sha256=tawggQMMHlYTGpnubxUAvDPDJe_Lpq-HpLLCSjbJV54,37320
 nextrec/basic/loggers.py,sha256=jJUzUt_kMpjpV2Mqr7qBENWA1olEutTI7dFnpmndUUw,13845
 nextrec/basic/metrics.py,sha256=nVz3AkKwsxj_M97CoZWyQj8_Y9ZM_Icvw_QCM6c33Bc,26262
-nextrec/basic/model.py,sha256=uKq74cZ7OvO6I9fHA3-ztpiD7xw-jevVKJicf-A_M9A,111999
+nextrec/basic/model.py,sha256=x4QZc8lNXHzpOJLeN3qPs9G_kNC1KgKn_2K52v3-vLw,132691
 nextrec/basic/session.py,sha256=mrIsjRJhmvcAfoO1pXX-KB3SK5CCgz89wH8XDoAiGEI,4475
 nextrec/basic/summary.py,sha256=hCDVB8127GSGtlfFnfEFHWXuvW5qjCSTwowNoA1i1xE,19815
 nextrec/data/__init__.py,sha256=YZQjpty1pDCM7q_YNmiA2sa5kbujUw26ObLHWjMPjKY,1194
 nextrec/data/batch_utils.py,sha256=TbnXYqYlmK51dJthaL6dO7LTn4wyp8221I-kdgvpvDE,3542
-nextrec/data/data_processing.py,sha256=lhuwYxWp4Ts2bbuLGDt2LmuPrOy7pNcKczd2uVcQ4ss,6476
+nextrec/data/data_processing.py,sha256=xD6afp4zc217ddKfDtHtToyDpxMDWvoqD_Vk4pIpvXU,6333
 nextrec/data/data_utils.py,sha256=0Ls1cnG9lBz0ovtyedw5vwp7WegGK_iF-F8e_3DEddo,880
 nextrec/data/dataloader.py,sha256=2sXwoiWxupKE-V1QYeZlXjK1yJyxhDtlOhknAnJF8Wk,19727
-nextrec/data/preprocessor.py,sha256=SsXkO5exQsXDZ2gx1Qwc4K58u0rG_K6lev8PbGipV58,68518
+nextrec/data/preprocessor.py,sha256=kOEfPy0t0M3jBA0kPIlwuSQYsuvn8yUNr_uE_NiulHU,49939
 nextrec/loss/__init__.py,sha256=rualGsY-IBvmM52q9eOBk0MyKcMkpkazcscOeDXi_SM,774
 nextrec/loss/grad_norm.py,sha256=I4jAs0f84I7MWmYZOMC0JRUNvBHZzhgpuox0hOtYWDg,7435
 nextrec/loss/listwise.py,sha256=mluxXQt9XiuWGvXA1nk4I0miqaKB6_GPVQqxLhAiJKs,5999
@@ -28,7 +28,6 @@ nextrec/models/generative/__init__.py,sha256=0MV3P-_ainPaTxmRBGWKUVCEt14KJvuvEHm
 nextrec/models/generative/tiger.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nextrec/models/multi_task/[pre]aitm.py,sha256=A2n0T4JEui-uHgbqITU5lpsmtnP14fQXRZM1peTPvhQ,6661
 nextrec/models/multi_task/[pre]snr_trans.py,sha256=k08tC-TI--a_Tt4_BmX0ZubzntyqwsejutYzbB5F4S4,9077
-nextrec/models/multi_task/[pre]star.py,sha256=BczXHPJtK7xyPbLO0fQ-w7qnzaBvLpyhG0CKMBUItCY,7057
 nextrec/models/multi_task/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nextrec/models/multi_task/apg.py,sha256=dNveyottpHTd811_3KVZQQqgffDzmX_rY1EMiv1oKeo,13536
 nextrec/models/multi_task/cross_stitch.py,sha256=o48ZZFKWXz-w4MWboVdxDxifM0V3_n2Is2nZ2HoJkfw,9441
@@ -41,15 +40,15 @@ nextrec/models/multi_task/ple.py,sha256=h8xPqd7BFM76GbQL6RzEbi2EAB8igrkFGS4zqhtW
 nextrec/models/multi_task/poso.py,sha256=jr-RaLl5UnZc1HcEIK3HrNnc_g17BImyJb43d4rEXpE,18218
 nextrec/models/multi_task/share_bottom.py,sha256=yM5--iqwEFQwpeQy_SmY8Vdo8a1Exi0-LNKSYeJz3hc,5238
 nextrec/models/ranking/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/models/ranking/afm.py,sha256=2IylgCi7DBZzF7bdWFWvsEBlnRmyradxdyJiz5P4Nwg,9397
+nextrec/models/ranking/afm.py,sha256=QPGBNlR9qjS5V5F-A8BnRBPcwwU6qcXQa96cNk9xlsY,9187
 nextrec/models/ranking/autoint.py,sha256=5g0tMtvkt3FiCjqmU76K7w7z-h3nMJbP0eVt5Ph4bbU,7259
 nextrec/models/ranking/dcn.py,sha256=qG3O2oL9ZLu9NBOJST06JeEh0k6UTXs-X1mQe_H4QCE,6679
 nextrec/models/ranking/dcn_v2.py,sha256=807wASeG56WYD7mEDEKaERw3r-Jpas6WhDzO0HGEk9I,10382
 nextrec/models/ranking/deepfm.py,sha256=1fuc-9f8DKDUH7EAY-XX5lITJ7Qw7ge6PTWFfQ7wld8,4404
-nextrec/models/ranking/dien.py,sha256=cluW44zOBHadf3oup6h7MpTexDXjH_VPkYcqsNBT80Y,18361
+nextrec/models/ranking/dien.py,sha256=9OM2vz1Umqlvny6UUySTEz5g_jf8I4creox9J7oV82A,18320
 nextrec/models/ranking/din.py,sha256=J6-S72_KJYLrzUmdrh6aAx-Qc1C08ZY7lY_KFtAkJz0,8855
 nextrec/models/ranking/eulernet.py,sha256=0nOBfccfvukSZLUNOCcB_azCh52DGJq-s9doyEGMN8E,11484
-nextrec/models/ranking/ffm.py,sha256=v15x2-rExcrEYdcPf2IxEgx-ImDSevhkhi4Oe4GbloY,10512
+nextrec/models/ranking/ffm.py,sha256=v8_whymddfY7u0F9rD4VTUQJXnyYYJYvbgwiR7DKuII,10432
 nextrec/models/ranking/fibinet.py,sha256=ejR1vNh5XM23SD7mfT686kuv3cmf5gKfkj0z_iMQqNA,7283
 nextrec/models/ranking/fm.py,sha256=SlFtbtnrZbeRnCHf-kUAMaeLV_wDgLiaBwPGeAO_ycM,3795
 nextrec/models/ranking/lr.py,sha256=0gmqPED-z7k4WVRy11WSLhYfS4bJQPRQTzQbe-rUITg,3227
@@ -63,9 +62,9 @@ nextrec/models/retrieval/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZ
 nextrec/models/retrieval/dssm.py,sha256=AmXDr62M6tivBg1P4MQ8f4cZnl1TGHxRBvZj05zWw64,6887
 nextrec/models/retrieval/dssm_v2.py,sha256=5ZH3dfNfRCDE69k8KG8BZJixaGOSVvQHB9uIDPMLPk4,5953
 nextrec/models/retrieval/mind.py,sha256=I0qVj39ApweRGW3qDNLca5vsNtJwRe7gBLh1pedsexY,14061
-nextrec/models/retrieval/sdm.py,sha256=1Y2gidG7WKuuGFaaQ8BcBGhQYoyyLPyhpRTo_xE1pmc,9987
+nextrec/models/retrieval/sdm.py,sha256=h9TqVmSJ8YF7hgPci784nAlBg1LazB641c4iEeuiLDg,9956
 nextrec/models/retrieval/youtube_dnn.py,sha256=hLyR4liuusJIjRg4vuaSoSEecYgDICipXnNFiA3o3oY,6351
-nextrec/models/sequential/hstu.py,sha256=iZcYLp44r23nHYNhGwD25JfH85DBrFwHOTg1WpHvLe8,18983
+nextrec/models/sequential/hstu.py,sha256=XFq-IERFg2ohqg03HkP6YinQaZUXljtYayUmvU-N_IY,18916
 nextrec/models/tree_base/__init__.py,sha256=ssGpU1viVidr1tgKCyvmkUXe32bUmwb5qPEkTh7ce70,342
 nextrec/models/tree_base/base.py,sha256=HcxUkctNgizY9LOBh2qXY6gUiYoy2JXA7j11NwUfWT4,26562
 nextrec/models/tree_base/catboost.py,sha256=hXINyx7iianwDxOZx3SLm0i-YP1jiC3HcAeqP9A2i4A,3434
@@ -78,10 +77,11 @@ nextrec/utils/data.py,sha256=pSL96mWjWfW_RKE-qlUSs9vfiYnFZAaRirzA6r7DB6s,24994
 nextrec/utils/embedding.py,sha256=akAEc062MG2cD7VIOllHaqtwzAirQR2gq5iW7oKpGAU,1449
 nextrec/utils/loss.py,sha256=GBWQGpDaYkMJySpdG078XbeUNXUC34PVqFy0AqNS9N0,4578
 nextrec/utils/model.py,sha256=PI9y8oWz1lhktgapZsiXb8rTr2NrFFlc80tr4yOFHik,5334
-nextrec/utils/torch_utils.py,sha256=fxViD6Pah0qnXtpvem6ncuLV7y58Q_gyktfvkZQo_JI,12207
+nextrec/utils/onnx_utils.py,sha256=KIVV_ELYzj3kCswfsSBZ1F2OnSwRJnXj7sxDBwBoBaA,8668
+nextrec/utils/torch_utils.py,sha256=_a9e6GXa3QKuu0E5RL44QRZ1iJSobbtNcPB3vtaCsu8,12313
 nextrec/utils/types.py,sha256=LFwYCBRo5WeYUh5LSCuyP1Lg9ez0Ih00Es3fUttGAFw,2273
-nextrec-0.4.34.dist-info/METADATA,sha256=2EpwNICyqgNVBUW2DNSMqzJR741TiCb-SrqZZx2VYtQ,23188
-nextrec-0.4.34.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-nextrec-0.4.34.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
-nextrec-0.4.34.dist-info/licenses/LICENSE,sha256=COP1BsqnEUwdx6GCkMjxOo5v3pUe4-Go_CdmQmSfYXM,1064
-nextrec-0.4.34.dist-info/RECORD,,
+nextrec-0.5.1.dist-info/METADATA,sha256=DswmHjwCt-xuCenbNsKJgeSMw2LqVOAJFVFjgEn2CRc,23532
+nextrec-0.5.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+nextrec-0.5.1.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
+nextrec-0.5.1.dist-info/licenses/LICENSE,sha256=COP1BsqnEUwdx6GCkMjxOo5v3pUe4-Go_CdmQmSfYXM,1064
+nextrec-0.5.1.dist-info/RECORD,,