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

nextrec/utils/model.py

@@ -2,7 +2,7 @@
 Model-related utilities for NextRec
 
 Date: create on 03/12/2025
-Checkpoint: edit on 29/12/2025
+Checkpoint: edit on 31/12/2025
 Author: Yang Zhou, zyaztec@gmail.com
 """
 
@@ -29,9 +29,9 @@ def merge_features(primary, secondary) -> list:
 
 
 def get_mlp_output_dim(params: dict, fallback: int) -> int:
-    dims = params.get("dims")
-    if dims:
-        return dims[-1]
+    hidden_dims = params.get("hidden_dims")
+    if hidden_dims:
+        return hidden_dims[-1]
     return fallback
 
 
@@ -72,47 +72,26 @@ 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:
+    default_losses = {
+        "pointwise": "bce",
+        "pairwise": "bpr",
+        "listwise": "listnet",
+    }
+    if 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)
+    elif isinstance(loss, list):
+        if len(loss) != nums_task:
+            raise ValueError(
+                f"[BaseModel-compile Error] Number of loss functions ({len(loss)}) must match number of tasks ({nums_task})."
+            )
+        loss_list = loss
     else:
-        loss_list = [effective_loss] * nums_task
+        loss_list = [loss] * nums_task
 
     for idx, mode in enumerate(training_modes):
         if isinstance(loss_list[idx], str) and loss_list[idx] in {
@@ -124,32 +103,6 @@ def get_loss_list(
     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/torch_utils.py

@@ -204,6 +204,11 @@ def get_scheduler(
             )
         else:
             raise NotImplementedError(f"Unsupported scheduler: {scheduler}")
+    elif isinstance(scheduler, type) and issubclass(
+        scheduler,
+        (torch.optim.lr_scheduler._LRScheduler, torch.optim.lr_scheduler.LRScheduler),
+    ):
+        scheduler_fn = scheduler(optimizer, **scheduler_params)
     elif isinstance(
         scheduler,
         (torch.optim.lr_scheduler._LRScheduler, torch.optim.lr_scheduler.LRScheduler),
@@ -215,6 +220,12 @@ def get_scheduler(
     return scheduler_fn
 
 
+def to_numpy(values: Any) -> np.ndarray:
+    if isinstance(values, torch.Tensor):
+        return values.detach().cpu().numpy()
+    return np.asarray(values)
+
+
 def to_tensor(
     value: Any, dtype: torch.dtype, device: torch.device | str | None = None
 ) -> torch.Tensor:

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextrec
-Version: 0.4.23
+Version: 0.4.25
 Summary: A comprehensive recommendation library with match, ranking, and multi-task learning models
 Project-URL: Homepage, https://github.com/zerolovesea/NextRec
 Project-URL: Repository, https://github.com/zerolovesea/NextRec
@@ -69,7 +69,7 @@ Description-Content-Type: text/markdown
 ![Python](https://img.shields.io/badge/Python-3.10+-blue.svg)
 ![PyTorch](https://img.shields.io/badge/PyTorch-1.10+-ee4c2c.svg)
 ![License](https://img.shields.io/badge/License-Apache%202.0-green.svg)
-![Version](https://img.shields.io/badge/Version-0.4.23-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.4.25-orange.svg)
 [![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
 
 中文文档 | [English Version](README_en.md)
@@ -182,7 +182,7 @@ sequence_features = [
     SequenceFeature(name='sequence_1', vocab_size=int(df['sequence_1'].apply(lambda x: max(x)).max() + 1), embedding_dim=16, padding_idx=0, embedding_name='sparse_0_emb'),]
 
 mlp_params = {
-    "dims": [256, 128, 64],
+    "hidden_dims": [256, 128, 64],
     "activation": "relu",
     "dropout": 0.3,
 }
@@ -249,11 +249,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
 
 预测结果固定保存到 `{checkpoint_path}/predictions/{name}.{save_data_format}`。
 
-> 截止当前版本0.4.23，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
+> 截止当前版本0.4.25，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
 
 ## 兼容平台
 
-当前最新版本为0.4.23，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
+当前最新版本为0.4.25，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
 
 | 平台 | 配置 | 
 |------|------|

{nextrec-0.4.23.dist-info → nextrec-0.4.25.dist-info}/RECORD

@@ -1,23 +1,23 @@
 nextrec/__init__.py,sha256=_M3oUqyuvQ5k8Th_3wId6hQ_caclh7M5ad51XN09m98,235
-nextrec/__version__.py,sha256=bUxoIOr-G9-PoGmh7zAW9CCJTt17Q0QuRmIjl2A39Sw,23
-nextrec/cli.py,sha256=Vm1XCFVw1vFh9NFw3PYZ_fYbh07tf45fl3RtPycooUI,24317
+nextrec/__version__.py,sha256=2KhS4HNlDzv_pmrC0ssutBbIG8FVsE-4OUVNs-FKmXw,23
+nextrec/cli.py,sha256=uOaXnlAM-ARrbxKOVWWkTE_rv-54px168kBhFUHtIAg,25073
 nextrec/basic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nextrec/basic/activation.py,sha256=uekcJsOy8SiT0_NaDO2VNSStyYFzVikDFVLDk-VrjwQ,2949
 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/layers.py,sha256=tr8XFOcTvUHEZ6T3zJwmtKMA-u_xfzHloIkItGs821U,40084
 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/metrics.py,sha256=CPzENDcpO6QTDZLBtQlfAGKUYYQc0FT-eaMKJ4MURFo,23396
+nextrec/basic/model.py,sha256=FFJIrMW0dqh89Jq1poXpNN-8XvPfcqIlKdChQpGH6x0,110083
 nextrec/basic/session.py,sha256=mrIsjRJhmvcAfoO1pXX-KB3SK5CCgz89wH8XDoAiGEI,4475
-nextrec/basic/summary.py,sha256=9xDtDbtMCPSQuEVLx23-SLL6qDRl1MfM19YMBG3Wtow,15372
+nextrec/basic/summary.py,sha256=b6jLo70gqZj_bQ4eb5yb8SXmr2ilZlKNN293EyVnkyc,17759
 nextrec/data/__init__.py,sha256=YZQjpty1pDCM7q_YNmiA2sa5kbujUw26ObLHWjMPjKY,1194
 nextrec/data/batch_utils.py,sha256=0bYGVX7RlhnHv_ZBaUngjDIpBNw-igCk98DgOsF7T6o,2879
-nextrec/data/data_processing.py,sha256=ZDZMSTBvxjPppl872is4M49o4WAkZXw2vUFOsNr0q3w,6658
+nextrec/data/data_processing.py,sha256=lhuwYxWp4Ts2bbuLGDt2LmuPrOy7pNcKczd2uVcQ4ss,6476
 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/data/dataloader.py,sha256=gTs4YC5tHHwTq0A9481KYK1XyloeN2dMVOjPAFehF_E,19972
+nextrec/data/preprocessor.py,sha256=AD5bHNbkAZAnI_SbDfJJaAh57CRtRjoOQJ6aIBkgoQs,65251
 nextrec/loss/__init__.py,sha256=rualGsY-IBvmM52q9eOBk0MyKcMkpkazcscOeDXi_SM,774
 nextrec/loss/grad_norm.py,sha256=YoE_XSIN1HOUcNq1dpfkIlWtMaB5Pu-SEWDaNgtRw1M,8316
 nextrec/loss/listwise.py,sha256=mluxXQt9XiuWGvXA1nk4I0miqaKB6_GPVQqxLhAiJKs,5999
@@ -26,16 +26,21 @@ nextrec/loss/pointwise.py,sha256=09nzI1L5eP9raXnj3Q49bD9Clp_JmsSWUvEj7bkTzSw,747
 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/multi_task/aitm.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nextrec/models/multi_task/apg.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nextrec/models/multi_task/cross_stitch.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nextrec/models/multi_task/esmm.py,sha256=sagHib2lqXya2fiAC8CdmUe-FJdcQKvFz6kziN42YRU,5724
+nextrec/models/multi_task/mmoe.py,sha256=16ZnaotUVxvYTSJp1Z1Zr3E1q3K2BdsmcYYCKhzaKWA,7830
+nextrec/models/multi_task/pepnet.py,sha256=GXKt6vQL-tMLFp0k7Nxbdfs-XNaB2Fi_qYHlnHKLlYo,12998
+nextrec/models/multi_task/ple.py,sha256=-ApSs_9cD0O5YbFE4MC0zP2Y5mjRRwl9I6CZ4eHpQQk,12408
+nextrec/models/multi_task/poso.py,sha256=QSE3yW0Y-Gb5c1YtBh5Z_xle5MiA6uRF4gyrEWPmMpI,18543
+nextrec/models/multi_task/share_bottom.py,sha256=mfP-_NjjYrTXAKFxZzA8LGNTc3paScTX3HbcQLKSoVg,5759
+nextrec/models/multi_task/snr_trans.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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/dcn_v2.py,sha256=GNfFSFlv6SFgXMUYSo53BD7fy0vkCMCN7EyrtN2DmlY,10169
 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
@@ -57,25 +62,25 @@ nextrec/models/representation/mf.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG
 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/retrieval/dssm.py,sha256=ZPNmjkP5TfdfIRyqrgFTGDqkGnKSjbmmjh8vxwtpfGM,7006
+nextrec/models/retrieval/dssm_v2.py,sha256=TV4t-W6n610T1dRwb4Ql71xDWUuM2TMjIO83KI79u0Y,6074
+nextrec/models/retrieval/mind.py,sha256=AKTSf77Ae7e3YWchuIAjbDv6xrLR3CbGNG7YtPBYVTk,14175
+nextrec/models/retrieval/sdm.py,sha256=LzUBec0Cd9YQUEDJJtQvue4DmfEzoh4hOOh91x1Dc6c,9647
+nextrec/models/retrieval/youtube_dnn.py,sha256=ciD9RyBy19mfQcEoqw1UfydmVBsJvffDw-sXA9kHRiI,6470
 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/config.py,sha256=WeGHjoQYA5SoC9B_uS3D6ChzKr9Z5n2fwE-8l6nsuLE,20425
 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/model.py,sha256=dcAL2lXNXRzFfCHMfOM_gIDLH68IAezMosSmmOD3FiQ,5624
+nextrec/utils/torch_utils.py,sha256=UQpWS7F3nITYqvx2KRBaQJc9oTowRkIvowhuQLt6NFM,11953
 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.25.dist-info/METADATA,sha256=AY7ejbF1WA7Z4EHCbZVKB-Flom6diV50LGwiUVt-auA,21859
+nextrec-0.4.25.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+nextrec-0.4.25.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
+nextrec-0.4.25.dist-info/licenses/LICENSE,sha256=2fQfVKeafywkni7MYHyClC6RGGC3laLTXCNBx-ubtp0,1064
+nextrec-0.4.25.dist-info/RECORD,,