nextrec 0.4.29__tar.gz → 0.4.31__tar.gz

{nextrec-0.4.29 → nextrec-0.4.31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextrec
-Version: 0.4.29
+Version: 0.4.31
 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.29-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.4.31-orange.svg)
 [![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
 
 中文文档 | [English Version](README_en.md)
@@ -254,11 +254,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
 
 预测结果固定保存到 `{checkpoint_path}/predictions/{name}.{save_data_format}`。
 
-> 截止当前版本0.4.29，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
+> 截止当前版本0.4.31，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
 
 ## 兼容平台
 
-当前最新版本为0.4.29，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
+当前最新版本为0.4.31，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
 
 | 平台 | 配置 | 
 |------|------|

{nextrec-0.4.29 → nextrec-0.4.31}/README.md

@@ -8,7 +8,7 @@
 ![Python](https://img.shields.io/badge/Python-3.10+-blue.svg)
 ![PyTorch](https://img.shields.io/badge/PyTorch-1.10+-ee4c2c.svg)
 ![License](https://img.shields.io/badge/License-Apache%202.0-green.svg)
-![Version](https://img.shields.io/badge/Version-0.4.29-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.4.31-orange.svg)
 [![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
 
 中文文档 | [English Version](README_en.md)
@@ -193,11 +193,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
 
 预测结果固定保存到 `{checkpoint_path}/predictions/{name}.{save_data_format}`。
 
-> 截止当前版本0.4.29，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
+> 截止当前版本0.4.31，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
 
 ## 兼容平台
 
-当前最新版本为0.4.29，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
+当前最新版本为0.4.31，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
 
 | 平台 | 配置 | 
 |------|------|

{nextrec-0.4.29 → nextrec-0.4.31}/README_en.md

@@ -8,7 +8,7 @@
 ![Python](https://img.shields.io/badge/Python-3.10+-blue.svg)
 ![PyTorch](https://img.shields.io/badge/PyTorch-1.10+-ee4c2c.svg)
 ![License](https://img.shields.io/badge/License-Apache%202.0-green.svg)
-![Version](https://img.shields.io/badge/Version-0.4.29-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.4.31-orange.svg)
 [![Ask DeepWiki](https://deepwiki.com/badge.svg)](https://deepwiki.com/zerolovesea/NextRec)
 
 English | [中文文档](README.md)
@@ -196,11 +196,11 @@ nextrec --mode=predict --predict_config=path/to/predict_config.yaml
 
 Prediction outputs are saved under `{checkpoint_path}/predictions/{name}.{save_data_format}`.
 
-> As of version 0.4.29, NextRec CLI supports single-machine training; distributed training features are currently under development.
+> As of version 0.4.31, NextRec CLI supports single-machine training; distributed training features are currently under development.
 
 ## Platform Compatibility
 
-The current version is 0.4.29. All models and test code have been validated on the following platforms. If you encounter compatibility issues, please report them in the issue tracker with your system version:
+The current version is 0.4.31. All models and test code have been validated on the following platforms. If you encounter compatibility issues, please report them in the issue tracker with your system version:
 
 | Platform | Configuration | 
 |----------|---------------|

{nextrec-0.4.29 → nextrec-0.4.31}/docs/rtd/conf.py

@@ -11,7 +11,7 @@ sys.path.insert(0, str(PROJECT_ROOT / "nextrec"))
 project = "NextRec"
 copyright = "2025, Yang Zhou"
 author = "Yang Zhou"
-release = "0.4.29"
+release = "0.4.31"
 
 extensions = [
     "myst_parser",

nextrec-0.4.31/nextrec/__version__.py

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

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec/cli.py

@@ -407,6 +407,8 @@ def train_model(train_config_path: str) -> None:
         use_swanlab=train_cfg.get("use_swanlab", False),
         wandb_api=train_cfg.get("wandb_api"),
         swanlab_api=train_cfg.get("swanlab_api"),
+        wandb_kwargs=train_cfg.get("wandb_kwargs"),
+        swanlab_kwargs=train_cfg.get("swanlab_kwargs"),
         log_interval=train_cfg.get("log_interval", 1),
         note=train_cfg.get("note"),
     )

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec/models/ranking/afm.py

@@ -59,6 +59,8 @@ class AFM(BaseModel):
         dense_features: list[DenseFeature] | None = None,
         sparse_features: list[SparseFeature] | None = None,
         sequence_features: list[SequenceFeature] | None = None,
+        target: str | list[str] | None = None,
+        task: str | list[str] | None = None,
         attention_dim: int = 32,
         attention_dropout: float = 0.0,
         **kwargs,
@@ -72,6 +74,8 @@ class AFM(BaseModel):
             dense_features=dense_features,
             sparse_features=sparse_features,
             sequence_features=sequence_features,
+            target=target,
+            task=task,
             **kwargs,
         )
 

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec/models/ranking/autoint.py

@@ -75,6 +75,8 @@ class AutoInt(BaseModel):
         dense_features: list[DenseFeature],
         sparse_features: list[SparseFeature],
         sequence_features: list[SequenceFeature],
+        target: str | list[str] | None = None,
+        task: str | list[str] | None = None,
         att_layer_num: int = 3,
         att_embedding_dim: int = 8,
         att_head_num: int = 2,
@@ -87,6 +89,8 @@ class AutoInt(BaseModel):
             dense_features=dense_features,
             sparse_features=sparse_features,
             sequence_features=sequence_features,
+            target=target,
+            task=task,
             **kwargs,
         )
 

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec/models/ranking/dcn.py

@@ -93,6 +93,8 @@ class DCN(BaseModel):
         dense_features: list[DenseFeature] | None = None,
         sparse_features: list[SparseFeature] | None = None,
         sequence_features: list[SequenceFeature] | None = None,
+        target: str | list[str] | None = None,
+        task: str | list[str] | None = None,
         cross_num: int = 3,
         mlp_params: dict | None = None,
         **kwargs,
@@ -107,6 +109,8 @@ class DCN(BaseModel):
             dense_features=dense_features,
             sparse_features=sparse_features,
             sequence_features=sequence_features,
+            target=target,
+            task=task,
             **kwargs,
         )
 

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec/models/ranking/dcn_v2.py

@@ -191,6 +191,8 @@ class DCNv2(BaseModel):
         dense_features: list[DenseFeature] | None = None,
         sparse_features: list[SparseFeature] | None = None,
         sequence_features: list[SequenceFeature] | None = None,
+        target: str | list[str] | None = None,
+        task: str | list[str] | None = None,
         cross_num: int = 3,
         cross_type: Literal["matrix", "mix", "low_rank"] = "matrix",
         architecture: Literal["parallel", "stacked"] = "parallel",
@@ -208,6 +210,8 @@ class DCNv2(BaseModel):
             dense_features=dense_features,
             sparse_features=sparse_features,
             sequence_features=sequence_features,
+            target=target,
+            task=task,
             **kwargs,
         )
 

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec/models/ranking/deepfm.py

@@ -61,6 +61,8 @@ class DeepFM(BaseModel):
         dense_features: list[DenseFeature] | None = None,
         sparse_features: list[SparseFeature] | None = None,
         sequence_features: list[SequenceFeature] | None = None,
+        target: str | list[str] | None = None,
+        task: str | list[str] | None = None,
         mlp_params: dict | None = None,
         **kwargs,
     ):
@@ -74,6 +76,8 @@ class DeepFM(BaseModel):
             dense_features=dense_features,
             sparse_features=sparse_features,
             sequence_features=sequence_features,
+            target=target,
+            task=task,
             **kwargs,
         )
 

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec/models/ranking/dien.py

@@ -203,6 +203,8 @@ class DIEN(BaseModel):
         dense_features: list[DenseFeature] | None = None,
         sparse_features: list[SparseFeature] | None = None,
         sequence_features: list[SequenceFeature] | None = None,
+        target: str | list[str] | None = None,
+        task: str | list[str] | None = None,
         behavior_feature_name: str | None = None,
         candidate_feature_name: str | None = None,
         neg_behavior_feature_name: str | None = None,
@@ -226,6 +228,8 @@ class DIEN(BaseModel):
             dense_features=dense_features,
             sparse_features=sparse_features,
             sequence_features=sequence_features,
+            target=target,
+            task=task,
             **kwargs,
         )
 

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec/models/ranking/din.py

@@ -72,6 +72,8 @@ class DIN(BaseModel):
         dense_features: list[DenseFeature] | None = None,
         sparse_features: list[SparseFeature] | None = None,
         sequence_features: list[SequenceFeature] | None = None,
+        target: str | list[str] | None = None,
+        task: str | list[str] | None = None,
         behavior_feature_name: str | None = None,
         candidate_feature_name: str | None = None,
         mlp_params: dict | None = None,
@@ -92,6 +94,8 @@ class DIN(BaseModel):
             dense_features=dense_features,
             sparse_features=sparse_features,
             sequence_features=sequence_features,
+            target=target,
+            task=task,
             **kwargs,
         )
 

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec/models/ranking/eulernet.py

@@ -199,6 +199,8 @@ class EulerNet(BaseModel):
         dense_features: list[DenseFeature] | None = None,
         sparse_features: list[SparseFeature] | None = None,
         sequence_features: list[SequenceFeature] | None = None,
+        target: str | list[str] | None = None,
+        task: str | list[str] | None = None,
         num_layers: int = 2,
         num_orders: int = 8,
         use_implicit: bool = True,
@@ -215,6 +217,8 @@ class EulerNet(BaseModel):
             dense_features=dense_features,
             sparse_features=sparse_features,
             sequence_features=sequence_features,
+            target=target,
+            task=task,
             **kwargs,
         )
 

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec/models/ranking/ffm.py

@@ -62,6 +62,8 @@ class FFM(BaseModel):
         dense_features: list[DenseFeature] | None = None,
         sparse_features: list[SparseFeature] | None = None,
         sequence_features: list[SequenceFeature] | None = None,
+        target: str | list[str] | None = None,
+        task: str | list[str] | None = None,
         **kwargs,
     ):
         dense_features = dense_features or []
@@ -72,6 +74,8 @@ class FFM(BaseModel):
             dense_features=dense_features,
             sparse_features=sparse_features,
             sequence_features=sequence_features,
+            target=target,
+            task=task,
             **kwargs,
         )
 

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec/models/ranking/fibinet.py

@@ -68,6 +68,8 @@ class FiBiNET(BaseModel):
         dense_features: list[DenseFeature] | None = None,
         sparse_features: list[SparseFeature] | None = None,
         sequence_features: list[SequenceFeature] | None = None,
+        target: str | list[str] | None = None,
+        task: str | list[str] | None = None,
         mlp_params: dict | None = None,
         interaction_combo: Literal[
             "01", "11", "10", "00"
@@ -88,6 +90,8 @@ class FiBiNET(BaseModel):
             dense_features=dense_features,
             sparse_features=sparse_features,
             sequence_features=sequence_features,
+            target=target,
+            task=task,
             **kwargs,
         )
 

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec/models/ranking/fm.py

@@ -59,6 +59,8 @@ class FM(BaseModel):
         dense_features: list[DenseFeature] | None = None,
         sparse_features: list[SparseFeature] | None = None,
         sequence_features: list[SequenceFeature] | None = None,
+        target: str | list[str] | None = None,
+        task: str | list[str] | None = None,
         **kwargs,
     ):
 
@@ -70,6 +72,8 @@ class FM(BaseModel):
             dense_features=dense_features,
             sparse_features=sparse_features,
             sequence_features=sequence_features,
+            target=target,
+            task=task,
             **kwargs,
         )
 

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec/models/ranking/lr.py

@@ -58,6 +58,8 @@ class LR(BaseModel):
         dense_features: list[DenseFeature] | None = None,
         sparse_features: list[SparseFeature] | None = None,
         sequence_features: list[SequenceFeature] | None = None,
+        target: str | list[str] | None = None,
+        task: str | list[str] | None = None,
         **kwargs,
     ):
 
@@ -69,6 +71,8 @@ class LR(BaseModel):
             dense_features=dense_features,
             sparse_features=sparse_features,
             sequence_features=sequence_features,
+            target=target,
+            task=task,
             **kwargs,
         )
 

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec/models/ranking/masknet.py

@@ -166,6 +166,8 @@ class MaskNet(BaseModel):
         dense_features: list[DenseFeature] | None = None,
         sparse_features: list[SparseFeature] | None = None,
         sequence_features: list[SequenceFeature] | None = None,
+        target: str | list[str] | None = None,
+        task: str | list[str] | None = None,
         architecture: Literal[
             "serial", "parallel"
         ] = "parallel",  # "serial" or "parallel"
@@ -185,6 +187,8 @@ class MaskNet(BaseModel):
             dense_features=dense_features,
             sparse_features=sparse_features,
             sequence_features=sequence_features,
+            target=target,
+            task=task,
             **kwargs,
         )
 

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec/models/ranking/pnn.py

@@ -57,6 +57,8 @@ class PNN(BaseModel):
         dense_features: list[DenseFeature] | None = None,
         sparse_features: list[SparseFeature] | None = None,
         sequence_features: list[SequenceFeature] | None = None,
+        target: str | list[str] | None = None,
+        task: str | list[str] | None = None,
         mlp_params: dict | None = None,
         product_type: Literal[
             "inner", "outer", "both"
@@ -76,6 +78,8 @@ class PNN(BaseModel):
             dense_features=dense_features,
             sparse_features=sparse_features,
             sequence_features=sequence_features,
+            target=target,
+            task=task,
             **kwargs,
         )
 

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec/models/ranking/widedeep.py

@@ -57,14 +57,20 @@ class WideDeep(BaseModel):
         dense_features: list[DenseFeature],
         sparse_features: list[SparseFeature],
         sequence_features: list[SequenceFeature],
-        mlp_params: dict,
+        target: str | list[str] | None = None,
+        task: str | list[str] | None = None,
+        mlp_params: dict | None = None,
         **kwargs,
     ):
 
+        mlp_params = mlp_params or {}
+
         super(WideDeep, self).__init__(
             dense_features=dense_features,
             sparse_features=sparse_features,
             sequence_features=sequence_features,
+            target=target,
+            task=task,
             **kwargs,
         )
 

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec/models/ranking/xdeepfm.py

@@ -116,18 +116,23 @@ class xDeepFM(BaseModel):
         dense_features: list[DenseFeature],
         sparse_features: list[SparseFeature],
         sequence_features: list[SequenceFeature],
-        mlp_params: dict,
+        target: str | list[str] | None = None,
+        task: str | list[str] | None = None,
+        mlp_params: dict | None = None,
         cin_size: list[int] | None = None,
         split_half: bool = True,
         **kwargs,
     ):
 
         cin_size = cin_size or [128, 128]
+        mlp_params = mlp_params or {}
 
         super(xDeepFM, self).__init__(
             dense_features=dense_features,
             sparse_features=sparse_features,
             sequence_features=sequence_features,
+            target=target,
+            task=task,
             **kwargs,
         )
 

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec_cli_preset/NextRec-CLI.md

@@ -119,6 +119,12 @@ train:
   use_swanlab: false                   # Enable SwanLab logging
   # wandb_api: YOUR_WANDB_API_KEY      # Optional API key for non-tty login
   # swanlab_api: YOUR_SWANLAB_API_KEY  # Optional API key for non-tty login
+  # wandb_kwargs:                      # Optional kwargs for wandb.init(...)
+  #   project: NextRec
+  #   name: example_run
+  # swanlab_kwargs:                    # Optional kwargs for swanlab.init(...)
+  #   project: NextRec
+  #   name: example_run
   device: cpu                          # Device: cpu, cuda, cuda:0, mps
 ```
 
@@ -185,6 +191,8 @@ train:
 - `use_swanlab`: Enable SwanLab logging
 - `wandb_api`: W&B API key for non-tty login
 - `swanlab_api`: SwanLab API key for non-tty login
+- `wandb_kwargs`: Optional kwargs for `wandb.init(...)`
+- `swanlab_kwargs`: Optional kwargs for `swanlab.init(...)`
 - `device`: Computing device
   - `cpu`: CPU
   - `cuda`: NVIDIA GPU

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec_cli_preset/NextRec-CLI_zh.md

@@ -119,6 +119,12 @@ train:
   use_swanlab: false                   # 启用 SwanLab 日志
   # wandb_api: YOUR_WANDB_API_KEY      # 可选，非交互登录用 API Key
   # swanlab_api: YOUR_SWANLAB_API_KEY  # 可选，非交互登录用 API Key
+  # wandb_kwargs:                      # 可选，wandb.init(...) 参数
+  #   project: NextRec
+  #   name: example_run
+  # swanlab_kwargs:                    # 可选，swanlab.init(...) 参数
+  #   project: NextRec
+  #   name: example_run
   device: cpu                          # 设备：cpu, cuda, cuda:0, mps
 ```
 
@@ -185,6 +191,8 @@ train:
 - `use_swanlab`: 启用 SwanLab 日志
 - `wandb_api`: W&B API key（非交互登录）
 - `swanlab_api`: SwanLab API key（非交互登录）
+- `wandb_kwargs`: `wandb.init(...)` 参数
+- `swanlab_kwargs`: `swanlab.init(...)` 参数
 - `device`: 运行设备
   - `cpu`: CPU
   - `cuda`: NVIDIA GPU

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec_cli_preset/train_config.yaml

@@ -42,6 +42,12 @@ train:
   # note: "baseline run"                         # Optional training note
   # wandb_api: YOUR_WANDB_API_KEY               # Optional API key for non-tty login
   # swanlab_api: YOUR_SWANLAB_API_KEY           # Optional API key for non-tty login
+  # wandb_kwargs:                               # Optional kwargs for wandb.init(...)
+  #   project: NextRec
+  #   name: example_run
+  # swanlab_kwargs:                             # Optional kwargs for swanlab.init(...)
+  #   project: NextRec
+  #   name: example_run
 
 # cd nextrec_cli_preset
 # nextrec --mode train --train_config train_config.yaml

{nextrec-0.4.29 → nextrec-0.4.31}/nextrec_cli_preset/train_config_template.yaml

@@ -160,6 +160,12 @@ train:
   use_swanlab: false                                   # Enable SwanLab logging
   # wandb_api: YOUR_WANDB_API_KEY                       # Optional API key for non-tty login
   # swanlab_api: YOUR_SWANLAB_API_KEY                   # Optional API key for non-tty login
+  # wandb_kwargs:                                      # Optional kwargs for wandb.init(...)
+  #   project: NextRec
+  #   name: example_run
+  # swanlab_kwargs:                                    # Optional kwargs for swanlab.init(...)
+  #   project: NextRec
+  #   name: example_run
   
   device: cpu                                          # Computation device
                                                        # Options:

{nextrec-0.4.29 → nextrec-0.4.31}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "nextrec"
-version = "0.4.29"
+version = "0.4.31"
 description = "A comprehensive recommendation library with match, ranking, and multi-task learning models"
 readme = "README.md"
 requires-python = ">=3.10"

nextrec-0.4.29/nextrec/__version__.py

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