nextrec 0.4.21__py3-none-any.whl → 0.4.23__py3-none-any.whl

nextrec/models/retrieval/mind.py

@@ -206,17 +206,6 @@ class MIND(BaseMatchModel):
         dense_l1_reg=0.0,
         embedding_l2_reg=0.0,
         dense_l2_reg=0.0,
-        optimizer: str | torch.optim.Optimizer = "adam",
-        optimizer_params: dict | None = None,
-        scheduler: (
-            str
-            | torch.optim.lr_scheduler._LRScheduler
-            | type[torch.optim.lr_scheduler._LRScheduler]
-            | None
-        ) = None,
-        scheduler_params: dict | None = None,
-        loss: str | nn.Module | list[str | nn.Module] | None = "bce",
-        loss_params: dict | list[dict] | None = None,
         **kwargs,
     ):
 
@@ -322,15 +311,6 @@ class MIND(BaseMatchModel):
             include_modules=["item_dnn"] if self.item_dnn else [],
         )
 
-        self.compile(
-            optimizer=optimizer,
-            optimizer_params=optimizer_params,
-            scheduler=scheduler,
-            scheduler_params=scheduler_params,
-            loss=loss,
-            loss_params=loss_params,
-        )
-
     def user_tower(self, user_input: dict) -> torch.Tensor:
         """
         User tower with multi-interest extraction

nextrec/models/retrieval/sdm.py

@@ -53,17 +53,6 @@ class SDM(BaseMatchModel):
         dense_l1_reg=0.0,
         embedding_l2_reg=0.0,
         dense_l2_reg=0.0,
-        optimizer: str | torch.optim.Optimizer = "adam",
-        optimizer_params: dict | None = None,
-        scheduler: (
-            str
-            | torch.optim.lr_scheduler._LRScheduler
-            | type[torch.optim.lr_scheduler._LRScheduler]
-            | None
-        ) = None,
-        scheduler_params: dict | None = None,
-        loss: str | nn.Module | list[str | nn.Module] | None = "bce",
-        loss_params: dict | list[dict] | None = None,
         **kwargs,
     ):
 
@@ -189,15 +178,6 @@ class SDM(BaseMatchModel):
             include_modules=["item_dnn"] if self.item_dnn else [],
         )
 
-        self.compile(
-            optimizer=optimizer,
-            optimizer_params=optimizer_params,
-            scheduler=scheduler,
-            scheduler_params=scheduler_params,
-            loss=loss,
-            loss_params=loss_params,
-        )
-
     def user_tower(self, user_input: dict) -> torch.Tensor:
         seq_feature = self.user_sequence_features[0]
         seq_input = user_input[seq_feature.name]

nextrec/models/retrieval/youtube_dnn.py

@@ -10,7 +10,6 @@ Reference:
 from typing import Literal
 
 import torch
-import torch.nn as nn
 
 from nextrec.basic.features import DenseFeature, SequenceFeature, SparseFeature
 from nextrec.basic.layers import MLP, EmbeddingLayer
@@ -54,17 +53,6 @@ class YoutubeDNN(BaseMatchModel):
         dense_l1_reg=0.0,
         embedding_l2_reg=0.0,
         dense_l2_reg=0.0,
-        optimizer: str | torch.optim.Optimizer = "adam",
-        optimizer_params: dict | None = None,
-        scheduler: (
-            str
-            | torch.optim.lr_scheduler._LRScheduler
-            | type[torch.optim.lr_scheduler._LRScheduler]
-            | None
-        ) = None,
-        scheduler_params: dict | None = None,
-        loss: str | nn.Module | list[str | nn.Module] | None = "bce",
-        loss_params: dict | list[dict] | None = None,
         **kwargs,
     ):
 
@@ -156,15 +144,6 @@ class YoutubeDNN(BaseMatchModel):
             embedding_attr="item_embedding", include_modules=["item_dnn"]
         )
 
-        self.compile(
-            optimizer=optimizer,
-            optimizer_params=optimizer_params,
-            scheduler=scheduler,
-            scheduler_params=scheduler_params,
-            loss=loss,
-            loss_params=loss_params,
-        )
-
     def user_tower(self, user_input: dict) -> torch.Tensor:
         """
         User tower to encode historical behavior sequences and user features.

nextrec/models/sequential/hstu.py

@@ -323,11 +323,6 @@ class HSTU(BaseModel):
         tie_embeddings: bool = True,
         target: Optional[list[str] | str] = None,
         task: str | list[str] | None = None,
-        optimizer: str = "adam",
-        optimizer_params: Optional[dict] = None,
-        scheduler: Optional[str] = None,
-        scheduler_params: Optional[dict] = None,
-        loss_params: Optional[dict] = None,
         embedding_l1_reg: float = 0.0,
         dense_l1_reg: float = 0.0,
         embedding_l2_reg: float = 0.0,
@@ -426,19 +421,6 @@ class HSTU(BaseModel):
         self.register_buffer("causal_mask", torch.empty(0), persistent=False)
         self.ignore_index = self.padding_idx if self.padding_idx is not None else -100
 
-        optimizer_params = optimizer_params or {}
-        scheduler_params = scheduler_params or {}
-        loss_params = loss_params or {}
-        loss_params.setdefault("ignore_index", self.ignore_index)
-
-        self.compile(
-            optimizer=optimizer,
-            optimizer_params=optimizer_params,
-            scheduler=scheduler,
-            scheduler_params=scheduler_params,
-            loss="crossentropy",
-            loss_params=loss_params,
-        )
         self.register_regularization_weights(
             embedding_attr="token_embedding",
             include_modules=["layers", "lm_head", "context_proj"],

nextrec/utils/__init__.py

@@ -6,7 +6,7 @@ Last update: 19/12/2025
 Author: Yang Zhou, zyaztec@gmail.com
 """
 
-from . import console, data, embedding, torch_utils
+from . import console, data, embedding, loss, torch_utils
 from .config import (
     build_feature_objects,
     build_model_instance,
@@ -38,6 +38,7 @@ from .data import (
 from .embedding import get_auto_embedding_dim
 from .feature import to_list
 from .model import compute_pair_scores, get_mlp_output_dim, merge_features
+from .loss import normalize_task_loss
 from .torch_utils import (
     add_distributed_sampler,
     get_device,
@@ -81,6 +82,8 @@ __all__ = [
     "merge_features",
     "get_mlp_output_dim",
     "compute_pair_scores",
+    # Loss utilities
+    "normalize_task_loss",
     # Feature utilities
     "to_list",
     # Config utilities
@@ -101,6 +104,7 @@ __all__ = [
     "console",
     "data",
     "embedding",
+    "loss",
     "torch_utils",
     # Type aliases
     "OptimizerName",

nextrec/{loss/loss_utils.py → utils/loss.py}

@@ -1,11 +1,13 @@
 """
 Loss utilities for NextRec.
 
-Date: create on 27/10/2025
-Checkpoint: edit on 19/12/2025
+Date: create on 28/12/2025
 Author: Yang Zhou, zyaztec@gmail.com
 """
 
+from __future__ import annotations
+
+import torch
 import torch.nn as nn
 
 from nextrec.loss.listwise import (
@@ -19,11 +21,19 @@ from nextrec.loss.pairwise import BPRLoss, HingeLoss, TripletLoss
 from nextrec.loss.pointwise import ClassBalancedFocalLoss, FocalLoss, WeightedBCELoss
 from nextrec.utils.types import LossName
 
-VALID_TASK_TYPES = [
-    "binary",
-    "multilabel",
-    "regression",
-]
+
+def normalize_task_loss(
+    task_loss,
+    valid_count,
+    total_count,
+    eps=1e-8,
+) -> torch.Tensor:
+    if not torch.is_tensor(valid_count):
+        valid_count = torch.tensor(float(valid_count), device=task_loss.device)
+    if not torch.is_tensor(total_count):
+        total_count = torch.tensor(float(total_count), device=task_loss.device)
+    scale = valid_count.to(task_loss.dtype) / (total_count.to(task_loss.dtype) + eps)
+    return task_loss * scale
 
 
 def build_cb_focal(kw):

nextrec/utils/model.py

@@ -2,7 +2,7 @@
 Model-related utilities for NextRec
 
 Date: create on 03/12/2025
-Checkpoint: edit on 24/12/2025
+Checkpoint: edit on 29/12/2025
 Author: Yang Zhou, zyaztec@gmail.com
 """
 
@@ -72,6 +72,84 @@ def compute_pair_scores(model, data, batch_size: int = 512):
     return scores.detach().cpu().numpy()
 
 
+def get_training_modes(
+    training_mode,
+    nums_task: int,
+    valid_modes: set[str] | None = None,
+) -> list:
+    valid_modes = valid_modes or {"pointwise", "pairwise", "listwise"}
+    if isinstance(training_mode, list):
+        training_modes = list(training_mode)
+        if len(training_modes) != nums_task:
+            raise ValueError(
+                "[BaseModel-init Error] training_mode list length must match number of tasks."
+            )
+    else:
+        training_modes = [training_mode] * nums_task
+    if any(mode not in valid_modes for mode in training_modes):
+        raise ValueError(
+            "[BaseModel-init Error] training_mode must be one of {'pointwise', 'pairwise', 'listwise'}."
+        )
+    return training_modes
+
+
+def get_loss_list(
+    loss,
+    training_modes: list[str],
+    nums_task: int,
+    default_losses: dict[str, str],
+):
+    effective_loss = loss
+    if effective_loss is None:
+        loss_list = [default_losses[mode] for mode in training_modes]
+    elif isinstance(effective_loss, list):
+        if not effective_loss:
+            loss_list = [default_losses[mode] for mode in training_modes]
+        else:
+            if len(effective_loss) != nums_task:
+                raise ValueError(
+                    f"[BaseModel-compile Error] Number of loss functions ({len(effective_loss)}) must match number of tasks ({nums_task})."
+                )
+            loss_list = list(effective_loss)
+    else:
+        loss_list = [effective_loss] * nums_task
+
+    for idx, mode in enumerate(training_modes):
+        if isinstance(loss_list[idx], str) and loss_list[idx] in {
+            "bce",
+            "binary_crossentropy",
+        }:
+            if mode in {"pairwise", "listwise"}:
+                loss_list[idx] = default_losses[mode]
+    return loss_list
+
+
+def resolve_loss_weights(loss_weights, nums_task: int):
+    if loss_weights is None:
+        return None
+    if nums_task == 1:
+        if isinstance(loss_weights, (list, tuple)):
+            if len(loss_weights) != 1:
+                raise ValueError(
+                    "[BaseModel-compile Error] loss_weights list must have exactly one element for single-task setup."
+                )
+            loss_weights = loss_weights[0]
+        return [float(loss_weights)]
+    if isinstance(loss_weights, (int, float)):
+        weights = [float(loss_weights)] * nums_task
+    elif isinstance(loss_weights, (list, tuple)):
+        weights = [float(w) for w in loss_weights]
+        if len(weights) != nums_task:
+            raise ValueError(
+                f"[BaseModel-compile Error] Number of loss_weights ({len(weights)}) must match number of tasks ({nums_task})."
+            )
+    else:
+        raise TypeError(
+            f"[BaseModel-compile Error] loss_weights must be int, float, list or tuple, got {type(loss_weights)}"
+        )
+    return weights
+
+
 def prepare_ranking_targets(
     y_pred: torch.Tensor, y_true: torch.Tensor
 ) -> tuple[torch.Tensor, torch.Tensor]:

nextrec/utils/types.py

@@ -34,26 +34,65 @@ LossName = Literal[
 ]
 
 ActivationName = Literal[
-        "dice",
-        "relu",
-        "relu6",
-        "elu",
-        "selu",
-        "leaky_relu",
-        "prelu",
-        "gelu",
-        "sigmoid",
-        "tanh",
-        "softplus",
-        "softsign",
-        "hardswish",
-        "mish",
-        "silu",
-        "swish",
-        "hardsigmoid",
-        "tanhshrink",
-        "softshrink",
-        "none",
-        "linear",
-        "identity",
-    ] 
+    "dice",
+    "relu",
+    "relu6",
+    "elu",
+    "selu",
+    "leaky_relu",
+    "prelu",
+    "gelu",
+    "sigmoid",
+    "tanh",
+    "softplus",
+    "softsign",
+    "hardswish",
+    "mish",
+    "silu",
+    "swish",
+    "hardsigmoid",
+    "tanhshrink",
+    "softshrink",
+    "none",
+    "linear",
+    "identity",
+]
+
+TrainingModeName = Literal["pointwise", "pairwise", "listwise"]
+
+TaskTypeName = Literal["binary", "regression"]
+
+MetricsName = Literal[
+    "auc",
+    "gauc",
+    "ks",
+    "logloss",
+    "accuracy",
+    "acc",
+    "precision",
+    "recall",
+    "f1",
+    "micro_f1",
+    "macro_f1",
+    "mse",
+    "mae",
+    "rmse",
+    "r2",
+    "mape",
+    "msle",
+    "auc",
+    "gauc",
+    "precision@10",
+    "hitrate@10",
+    "map@10",
+    "cosine",
+    "recall@5",
+    "recall@10",
+    "recall@20",
+    "ndcg@5",
+    "ndcg@10",
+    "ndcg@20",
+    "mrr@5",
+    "mrr@10",
+    "mrr@20",
+]

{nextrec-0.4.21.dist-info → nextrec-0.4.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextrec
-Version: 0.4.21
+Version: 0.4.23
 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
@@ -65,11 +65,11 @@ Description-Content-Type: text/markdown
 
 <div align="center">
 
-[![PyPI Downloads](https://static.pepy.tech/personalized-badge/nextrec?period=total&units=NONE&left_color=BLACK&right_color=GREEN&left_text=PyPI-downloads)](https://pypistats.org/packages/nextrec)
+[![PyPI Downloads](https://static.pepy.tech/personalized-badge/nextrec?period=total&units=NONE&left_color=grey&right_color=GREEN&left_text=PyPI-downloads)](https://pypistats.org/packages/nextrec)
 ![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.21-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.4.23-orange.svg)
 [![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
 
 中文文档 | [English Version](README_en.md)
@@ -191,6 +191,8 @@ model = DIN(
     dense_features=dense_features,
     sparse_features=sparse_features,
     sequence_features=sequence_features,
+    behavior_feature_name="sequence_0",
+    candidate_feature_name="item_id",
     mlp_params=mlp_params,
     attention_hidden_units=[80, 40],
     attention_activation='sigmoid',
@@ -204,7 +206,7 @@ model = DIN(
     session_id="din_tutorial",                            # 实验id，用于存放训练日志
 )
 
-# 编译模型，设置优化器和损失函数
+# 编译模型，优化器/损失/学习率调度器统一在 compile 中设置
 model.compile(
             optimizer = "adam",
             optimizer_params = {"lr": 1e-3, "weight_decay": 1e-5},
@@ -247,11 +249,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
 
 预测结果固定保存到 `{checkpoint_path}/predictions/{name}.{save_data_format}`。
 
-> 截止当前版本0.4.21，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
+> 截止当前版本0.4.23，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
 
 ## 兼容平台
 
-当前最新版本为0.4.21，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
+当前最新版本为0.4.23，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
 
 | 平台 | 配置 | 
 |------|------|

nextrec-0.4.23.dist-info/RECORD

@@ -0,0 +1,81 @@
+nextrec/__init__.py,sha256=_M3oUqyuvQ5k8Th_3wId6hQ_caclh7M5ad51XN09m98,235
+nextrec/__version__.py,sha256=bUxoIOr-G9-PoGmh7zAW9CCJTt17Q0QuRmIjl2A39Sw,23
+nextrec/cli.py,sha256=Vm1XCFVw1vFh9NFw3PYZ_fYbh07tf45fl3RtPycooUI,24317
+nextrec/basic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nextrec/basic/activation.py,sha256=uekcJsOy8SiT0_NaDO2VNSStyYFzVikDFVLDk-VrjwQ,2949
+nextrec/basic/callback.py,sha256=7geza5iMMlMojlrIKH5A7nzvCe4IYwgUaMRh_xpblWk,12585
+nextrec/basic/features.py,sha256=zLijBNkKwCXv9TKxSWwvmt7aVfWn2D5JvfwukeIRqec,9174
+nextrec/basic/heads.py,sha256=BshykLxD41KxKuZaBxf4Fmy1Mc52b3ioJliN1BVaGlk,3374
+nextrec/basic/layers.py,sha256=74RZiyYgiY9YFb2hWWNEBdWjvx2bXzCF3WtJJeSDtXQ,37857
+nextrec/basic/loggers.py,sha256=KxTPVHtkebAbpxZIYZ4aqncZCu-dccpKtIxmi2bVs6o,13160
+nextrec/basic/metrics.py,sha256=CkdMOq_RsQKd9qBGVWsNI9UF16yK4N-SnDvmkwA9KeY,23076
+nextrec/basic/model.py,sha256=bj60PWmG3wl9xmuZAwpnEB7XkBAr5kkZE-UA3Z5iaxU,103976
+nextrec/basic/session.py,sha256=mrIsjRJhmvcAfoO1pXX-KB3SK5CCgz89wH8XDoAiGEI,4475
+nextrec/basic/summary.py,sha256=9xDtDbtMCPSQuEVLx23-SLL6qDRl1MfM19YMBG3Wtow,15372
+nextrec/data/__init__.py,sha256=YZQjpty1pDCM7q_YNmiA2sa5kbujUw26ObLHWjMPjKY,1194
+nextrec/data/batch_utils.py,sha256=0bYGVX7RlhnHv_ZBaUngjDIpBNw-igCk98DgOsF7T6o,2879
+nextrec/data/data_processing.py,sha256=ZDZMSTBvxjPppl872is4M49o4WAkZXw2vUFOsNr0q3w,6658
+nextrec/data/data_utils.py,sha256=0Ls1cnG9lBz0ovtyedw5vwp7WegGK_iF-F8e_3DEddo,880
+nextrec/data/dataloader.py,sha256=43eTLqhWKcJdrFiGzlrz8zIgLAQsylRQ_DOklPNmKr4,18993
+nextrec/data/preprocessor.py,sha256=4mVhQ6W2M9nmTeQjArx_cndWwnk2i29U2iXSNgg5gXM,52917
+nextrec/loss/__init__.py,sha256=rualGsY-IBvmM52q9eOBk0MyKcMkpkazcscOeDXi_SM,774
+nextrec/loss/grad_norm.py,sha256=YoE_XSIN1HOUcNq1dpfkIlWtMaB5Pu-SEWDaNgtRw1M,8316
+nextrec/loss/listwise.py,sha256=mluxXQt9XiuWGvXA1nk4I0miqaKB6_GPVQqxLhAiJKs,5999
+nextrec/loss/pairwise.py,sha256=9fyH9p2u-N0-jAnNTq3X5Dje0ipj1dob8wp-yQKRra4,3493
+nextrec/loss/pointwise.py,sha256=09nzI1L5eP9raXnj3Q49bD9Clp_JmsSWUvEj7bkTzSw,7474
+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=pZXK7WQVR3UUV8AKPZUHQnk7KxIA4Yp4Gmt5H8rsSu8,5734
+nextrec/models/multi_task/mmoe.py,sha256=vihfiWkNFLyBF7juimUzq9Sg2id4ExD7ShFtN7RerOc,7817
+nextrec/models/multi_task/ple.py,sha256=Mf_RPtENCjj0WgTm0TDL5blZZuph8XWi9y-M36TvNBY,12362
+nextrec/models/multi_task/poso.py,sha256=O2K4nFRk0Lm-YCYCK258iwxzYRh7tdJDACm_87cFE10,18427
+nextrec/models/multi_task/share_bottom.py,sha256=aX449O09qlN5D19HCYAa4uwKa1hq00NJs4uLriJzqM0,5746
+nextrec/models/ranking/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nextrec/models/ranking/afm.py,sha256=rTuLH_6GD8DXEu5GjuypdAXpyc-Ug4A2pIQ9y5mlA24,9177
+nextrec/models/ranking/autoint.py,sha256=UVuVyinTxy29gw_0bI9aP6xYzg3fpDfN-11JjqgnM68,7026
+nextrec/models/ranking/dcn.py,sha256=elRw4qEE9d8tYmZEqdGlE_l-_fQ5CacpVY1IrbyX6iQ,6466
+nextrec/models/ranking/dcn_v2.py,sha256=XhSz3xZOTen6hrsW8eV9GwnEJ7etpnyUV_sTPmqUzxM,10172
+nextrec/models/ranking/deepfm.py,sha256=Y6gVcZzHeOxUAT5QT2I2y8hlrL0A9gG3L92xCshPs9E,4191
+nextrec/models/ranking/dien.py,sha256=QtinHyg6kvbJPhPTC1KUTOngf8HFcsuEMBu3yyxOU2Y,18116
+nextrec/models/ranking/din.py,sha256=kh9oAAq8WvLCoTtI0Im4gUZIlK8LC7jLgGclq-ucW3s,8605
+nextrec/models/ranking/eulernet.py,sha256=pcZLOj0h97jp32vXaZeuDku82t1Qsa3NFjPimTGq48Q,11274
+nextrec/models/ranking/ffm.py,sha256=-cAfSB6eU3-P2ZXv9P4OmsetFt1X7zYO83wBz6uR6is,10302
+nextrec/models/ranking/fibinet.py,sha256=IapVFh35j95FKyHGr6VNSavFJQ_FMXLzpb_r24xHnaM,7068
+nextrec/models/ranking/fm.py,sha256=Oby9rikk2-V8xKLeg3hMPFH10V0rCiYA14pmbn9dqe8,3585
+nextrec/models/ranking/lr.py,sha256=jaWd1V7PAPvra7M1eStAI_GFXqc2eKeWRVJ4MvlSp7g,3017
+nextrec/models/ranking/masknet.py,sha256=u0rVHfRQpskCyPJdAC0D7ySf2PkhNBCEMmAce-JvlfA,11396
+nextrec/models/ranking/pnn.py,sha256=vxx5jnrWeerXKet06aYa74edEjly4BYRcGsDYgSp38A,7338
+nextrec/models/ranking/widedeep.py,sha256=8kF695q-gR_ECfKkspYgDEbWkh_Unnt5Ta3GuqbxnzQ,3999
+nextrec/models/ranking/xdeepfm.py,sha256=BjuZYiIkpMAevJZWx_NUgtFfLwamwuLXuarUpEidTsc,7240
+nextrec/models/representation/__init__.py,sha256=O3QHMMXBszwM-mTl7bA3wawNZvDGet-QIv6Ys5GHGJ8,190
+nextrec/models/representation/autorec.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nextrec/models/representation/bpr.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nextrec/models/representation/cl4srec.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nextrec/models/representation/lightgcn.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nextrec/models/representation/mf.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nextrec/models/representation/rqvae.py,sha256=JyZxVY9CibcdBGk97TxjG5O3WQC10_60tHNcP_qtegs,29290
+nextrec/models/representation/s3rec.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nextrec/models/retrieval/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nextrec/models/retrieval/dssm.py,sha256=NxqGYkbRhJ_tXs8ipklEmQcsyz9Jyh21Hkv5dgoU2wk,6998
+nextrec/models/retrieval/dssm_v2.py,sha256=AeOn5XZgqImqRWofEDmHvoc65iFH5UrbP_k9sffx97M,6066
+nextrec/models/retrieval/mind.py,sha256=oF79m6bZ9yAKCTB_IJL545J11cU2izKO2q-1ifI4TNA,14171
+nextrec/models/retrieval/sdm.py,sha256=8c9hB8g4my5YYE178wnicmICX5Q3PLU4Z-E2GUDrHFE,9639
+nextrec/models/retrieval/youtube_dnn.py,sha256=BkEr1jGBnsogFJ1LtgC4oytCqCuP-iX9i_NvUUP9RKM,6462
+nextrec/models/sequential/hstu.py,sha256=4-EUOQ4HTRG5MAhTA2b9FOOXXw8oyPxDBaaDFunkT6o,18979
+nextrec/models/sequential/sasrec.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nextrec/utils/__init__.py,sha256=jD73RLigxcHFP-rXBoPi2VUTKH7kE5vNMQkr4lW8UUY,2655
+nextrec/utils/config.py,sha256=UIi4zntP2g4IJaeMQYoa6kMQlU_23Hq4N1ZugMgnB5A,20331
+nextrec/utils/console.py,sha256=RA3ZTjtUQXvueouSmXJNLkRjeUGQZesphwWjFMTbV4I,13577
+nextrec/utils/data.py,sha256=pSL96mWjWfW_RKE-qlUSs9vfiYnFZAaRirzA6r7DB6s,24994
+nextrec/utils/embedding.py,sha256=akAEc062MG2cD7VIOllHaqtwzAirQR2gq5iW7oKpGAU,1449
+nextrec/utils/feature.py,sha256=E3NOFIW8gAoRXVrDhCSonzg8k7nMUZyZzMfCq9k73_A,623
+nextrec/utils/loss.py,sha256=GBWQGpDaYkMJySpdG078XbeUNXUC34PVqFy0AqNS9N0,4578
+nextrec/utils/model.py,sha256=UmDdV23ra7klvZZ3HEbRWBEeMrKljNoeq3hSVWT6a6o,7558
+nextrec/utils/torch_utils.py,sha256=1lvZ7BG-rGLIAlumQIoeq5T9dO9hx2p8sa2_DC_bTZU,11564
+nextrec/utils/types.py,sha256=VhtLXUVvu0zAZVAUgRUML4FExRC-GH-ZmC1UiVSr3HE,1523
+nextrec-0.4.23.dist-info/METADATA,sha256=y2SWDa4fk2I1SnlLr4Svb6FTHp1oxvZrleL4OAlxuos,21852
+nextrec-0.4.23.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+nextrec-0.4.23.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
+nextrec-0.4.23.dist-info/licenses/LICENSE,sha256=2fQfVKeafywkni7MYHyClC6RGGC3laLTXCNBx-ubtp0,1064
+nextrec-0.4.23.dist-info/RECORD,,

nextrec-0.4.21.dist-info/RECORD

@@ -1,81 +0,0 @@
-nextrec/__init__.py,sha256=_M3oUqyuvQ5k8Th_3wId6hQ_caclh7M5ad51XN09m98,235
-nextrec/__version__.py,sha256=HRnyuWvhHjVmMtjn-Iir0_ZbxVK3sjpTYkAqLIRVMIE,23
-nextrec/cli.py,sha256=Vm1XCFVw1vFh9NFw3PYZ_fYbh07tf45fl3RtPycooUI,24317
-nextrec/basic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/basic/activation.py,sha256=HfSNRXcsCFvPRxVUuV4s_NDoltE1CC_e40-9fxtnrMA,2975
-nextrec/basic/callback.py,sha256=7geza5iMMlMojlrIKH5A7nzvCe4IYwgUaMRh_xpblWk,12585
-nextrec/basic/features.py,sha256=zLijBNkKwCXv9TKxSWwvmt7aVfWn2D5JvfwukeIRqec,9174
-nextrec/basic/heads.py,sha256=yKLFzYCyNVHoQ2JUNXdEiX1JldwZODl4nXDmwh0xVYo,3391
-nextrec/basic/layers.py,sha256=74RZiyYgiY9YFb2hWWNEBdWjvx2bXzCF3WtJJeSDtXQ,37857
-nextrec/basic/loggers.py,sha256=KxTPVHtkebAbpxZIYZ4aqncZCu-dccpKtIxmi2bVs6o,13160
-nextrec/basic/metrics.py,sha256=1r6efTc9TpARNBt5X9ISoppTZflej6EdFkjPYHV-YZI,23162
-nextrec/basic/model.py,sha256=nKuDy64ebKV4yB9qp1iIThDQoxUDX-U-nSSiLNu-hnc,103836
-nextrec/basic/session.py,sha256=mrIsjRJhmvcAfoO1pXX-KB3SK5CCgz89wH8XDoAiGEI,4475
-nextrec/basic/summary.py,sha256=I6vNc-W_mVo7EogsFUTXf20bWYMVnTab60Zs6wSxsdc,14406
-nextrec/data/__init__.py,sha256=YZQjpty1pDCM7q_YNmiA2sa5kbujUw26ObLHWjMPjKY,1194
-nextrec/data/batch_utils.py,sha256=0bYGVX7RlhnHv_ZBaUngjDIpBNw-igCk98DgOsF7T6o,2879
-nextrec/data/data_processing.py,sha256=ZDZMSTBvxjPppl872is4M49o4WAkZXw2vUFOsNr0q3w,6658
-nextrec/data/data_utils.py,sha256=0Ls1cnG9lBz0ovtyedw5vwp7WegGK_iF-F8e_3DEddo,880
-nextrec/data/dataloader.py,sha256=43eTLqhWKcJdrFiGzlrz8zIgLAQsylRQ_DOklPNmKr4,18993
-nextrec/data/preprocessor.py,sha256=pilzqbluAn1QykeBVPxvnQcbRUuZr3aX9hCQqey--Ks,47245
-nextrec/loss/__init__.py,sha256=ZCgsfyR5YAecv6MdOsnUjkfacvZg2coQVjuKAfPvmRo,923
-nextrec/loss/grad_norm.py,sha256=1BU1uHh6CuNRc_M_bPP2mrVKOnUGQWv_tR_8-ETOJlg,8385
-nextrec/loss/listwise.py,sha256=UT9vJCOTOQLogVwaeTV7Z5uxIYnngGdxk-p9e97MGkU,5744
-nextrec/loss/loss_utils.py,sha256=Bl3PJ-AQrTDlV_uGJV4M6XCgmf8X2Z0h4nAP9o40ngU,4168
-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=nDd_Hr3tBdhDs77O5WlN5tafi4eizAvaGYdBqhl3_ws,6480
-nextrec/models/multi_task/mmoe.py,sha256=nMAIOp9_wXvQTZqRUHRIUSSCxSkH9L48oqe6MB_d2Xk,8614
-nextrec/models/multi_task/ple.py,sha256=Pzj7Z_1rG85rlHgvuBZ4kOa_emSZd94KjalDzmi9uuo,13050
-nextrec/models/multi_task/poso.py,sha256=vOBU4ZW22P8PBoN0isuGbAbLhHa_rMpKUSL5zs-HTvM,19167
-nextrec/models/multi_task/share_bottom.py,sha256=NUzoL_A6XL2ZHiB3C-VfnBAuver_Bu9tty7nr3QwpjQ,6549
-nextrec/models/ranking/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/models/ranking/afm.py,sha256=_TdRIUtgx2jqOgamisZTekesGUUhdxvLWR9SJ0NZZF4,10082
-nextrec/models/ranking/autoint.py,sha256=_nA0umo-PKrwLh1g_6HcbQLWakKpsAxP7G1jI1-bL1k,8066
-nextrec/models/ranking/dcn.py,sha256=LVoOzc6ibHRntLX_Gtlz7b18tpX_CCyg7448tvG4KMA,7327
-nextrec/models/ranking/dcn_v2.py,sha256=9jWND-OVPtGfn_aJxuvuuAUJFJ48JD89khb-_mVQnVY,11120
-nextrec/models/ranking/deepfm.py,sha256=l57dAwtolWWBLvkCsZFZeOkWHxMbBZateBGKYbIKPQ0,5144
-nextrec/models/ranking/dien.py,sha256=hVRq4OUJ4UM3Q8BC3tkasAJSRdoa_MCPpU4DvhxlMhw,18982
-nextrec/models/ranking/din.py,sha256=tFBQ-3JcJcQde6kQPDTG9gMWnq7hVEgSj-x5r02XLvc,9446
-nextrec/models/ranking/eulernet.py,sha256=RReaGKLDdX0xQMxm-O44d3FU0bpja24x-Chl46-woOE,12140
-nextrec/models/ranking/ffm.py,sha256=2m3cfbXgSU4IUdGu-iteqYgAurM72Ens3hnrIZPaylU,11201
-nextrec/models/ranking/fibinet.py,sha256=9zN4tdgdzfLksb6BPl6Pcm4mqcHAXQHTJz8WTvFXO4w,7872
-nextrec/models/ranking/fm.py,sha256=A9hzs3RW6GjQxwKirEBDCus68Ve5jfSwdxuzGw-KZno,4493
-nextrec/models/ranking/lr.py,sha256=Qy83WSbv6_maCrdehUvRlWfql7LszFEZlvxzjxL28yw,3965
-nextrec/models/ranking/masknet.py,sha256=D2qGSplXqlc7Y4nt6tgKHyufLkVglijEsPYSoybYyHA,12276
-nextrec/models/ranking/pnn.py,sha256=Tt7kNUBIgU62FqVxK0BBFoanuQpwgKyhUvJTz1e06I8,8159
-nextrec/models/ranking/widedeep.py,sha256=ZdJbQQT70Q49sGtgCKGOEtVlKCW4uASf1JxOWBhP-oA,4998
-nextrec/models/ranking/xdeepfm.py,sha256=fHIRJj6Ai1BnEooaYt6JLbdyE7A_0P0aqH_0Tc4VdEE,8216
-nextrec/models/representation/__init__.py,sha256=O3QHMMXBszwM-mTl7bA3wawNZvDGet-QIv6Ys5GHGJ8,190
-nextrec/models/representation/autorec.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/models/representation/bpr.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/models/representation/cl4srec.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/models/representation/lightgcn.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/models/representation/mf.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/models/representation/rqvae.py,sha256=JyZxVY9CibcdBGk97TxjG5O3WQC10_60tHNcP_qtegs,29290
-nextrec/models/representation/s3rec.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/models/retrieval/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/models/retrieval/dssm.py,sha256=LGOhXH2pOcaQyabWLUvbSTuDOKOM3vaV4hrQ4LgF_M8,7834
-nextrec/models/retrieval/dssm_v2.py,sha256=3fvsLzEBo9fJ75YKoGgh4nTvVT3z5rE1Mpnmd8xP5Nc,6902
-nextrec/models/retrieval/mind.py,sha256=ZiXK-sMQulXK_ifVzIp4V1MW6manga8BDZXcMCwFbKU,14881
-nextrec/models/retrieval/sdm.py,sha256=vNsfqmIP5XwuUJve8ettCByXfHloq6BM7SHYeIr2ss8,10349
-nextrec/models/retrieval/youtube_dnn.py,sha256=ck4ja8v_BXG7X6AoEU_bUExvSHCWJBo-fNw3Y4zz9IQ,7194
-nextrec/models/sequential/hstu.py,sha256=tlZR-UMhY5dMQVqmWYdkUg54h5W3vQUctwHA5TyThMo,19664
-nextrec/models/sequential/sasrec.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/utils/__init__.py,sha256=Fb5yjSHBsKBitRuJKbhEbQDj3WMTGgjb3dazTwXezo8,2551
-nextrec/utils/config.py,sha256=UIi4zntP2g4IJaeMQYoa6kMQlU_23Hq4N1ZugMgnB5A,20331
-nextrec/utils/console.py,sha256=RA3ZTjtUQXvueouSmXJNLkRjeUGQZesphwWjFMTbV4I,13577
-nextrec/utils/data.py,sha256=pSL96mWjWfW_RKE-qlUSs9vfiYnFZAaRirzA6r7DB6s,24994
-nextrec/utils/embedding.py,sha256=akAEc062MG2cD7VIOllHaqtwzAirQR2gq5iW7oKpGAU,1449
-nextrec/utils/feature.py,sha256=E3NOFIW8gAoRXVrDhCSonzg8k7nMUZyZzMfCq9k73_A,623
-nextrec/utils/model.py,sha256=fHvFciUuMOVcM1oWiRva4LcArRdZ1R5Uzml-COSqqvM,4688
-nextrec/utils/torch_utils.py,sha256=1lvZ7BG-rGLIAlumQIoeq5T9dO9hx2p8sa2_DC_bTZU,11564
-nextrec/utils/types.py,sha256=cLHgo0Nd69t5cHcQrPIDNwtKKhrcbghS4XvXbs5P1xs,1036
-nextrec-0.4.21.dist-info/METADATA,sha256=BggHbRJHkiNpRBDoZJDR_waUDcvlXW-EWkqZSbhLGBw,21743
-nextrec-0.4.21.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-nextrec-0.4.21.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
-nextrec-0.4.21.dist-info/licenses/LICENSE,sha256=2fQfVKeafywkni7MYHyClC6RGGC3laLTXCNBx-ubtp0,1064
-nextrec-0.4.21.dist-info/RECORD,,