nextrec 0.4.29__py3-none-any.whl → 0.4.31__py3-none-any.whl

nextrec/__version__.py

@@ -1 +1 @@
-__version__ = "0.4.29"
+__version__ = "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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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.dist-info → nextrec-0.4.31.dist-info}/METADATA

@@ -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.dist-info → nextrec-0.4.31.dist-info}/RECORD

@@ -1,6 +1,6 @@
 nextrec/__init__.py,sha256=_M3oUqyuvQ5k8Th_3wId6hQ_caclh7M5ad51XN09m98,235
-nextrec/__version__.py,sha256=LtSKwkqgueDQK6uLnhs8HgKylJHH36EMo48UDeHgv2Y,23
-nextrec/cli.py,sha256=uOaXnlAM-ARrbxKOVWWkTE_rv-54px168kBhFUHtIAg,25073
+nextrec/__version__.py,sha256=3w2Z9VwNeAAXivgSAMMwqgJrAaUllRP_T-sKqxgdcns,23
+nextrec/cli.py,sha256=WDiJqEzDLRCP8S_0EsBWgVAg_G2rF7mPNGfdlhZ-nO4,25181
 nextrec/basic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nextrec/basic/activation.py,sha256=uekcJsOy8SiT0_NaDO2VNSStyYFzVikDFVLDk-VrjwQ,2949
 nextrec/basic/asserts.py,sha256=U1EKovV_OT7_Mm99zFvdfF2hccFREp3gdDaeRjfiBwQ,2249
@@ -41,22 +41,22 @@ nextrec/models/multi_task/ple.py,sha256=cO-NqEm-UZKRz2MznBjqsXL8ImH7WU1HRzXdWAtb
 nextrec/models/multi_task/poso.py,sha256=Xjw9JBAiGR9CGewp1uS4b1soA7fOvSWTsIT6pS9_o30,18215
 nextrec/models/multi_task/share_bottom.py,sha256=BT-nu0NZTV4HlFkva_KnoKLSxB0-gYuJWPw7PRDGwC8,5172
 nextrec/models/ranking/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/models/ranking/afm.py,sha256=CNiWdGcp9pQiEBbv10HPT5I2NnJlXqvEHZKIm-J-Q44,9189
-nextrec/models/ranking/autoint.py,sha256=q2C1pGZxQ0SchYMJWzD3EOeaiU3Kp7Xn57-Iidwj4IA,7051
-nextrec/models/ranking/dcn.py,sha256=fD0xBGDYtytGlfnNX6GNMkim4DDlZa-JutLu7q9L_hU,6471
-nextrec/models/ranking/dcn_v2.py,sha256=39mZPr33xGgE9zOJfoZceWv993LG9EqsNW4W8xS0duc,10174
-nextrec/models/ranking/deepfm.py,sha256=xAPjoZnam4T2yvBE07VAeqZt3vJeblOxyAAeTerBnRo,4196
-nextrec/models/ranking/dien.py,sha256=OhLp5XiXIZqyP47H7dfdMRZIIyTQVGWZqTt48wdOLxY,18153
-nextrec/models/ranking/din.py,sha256=ezhw1N35537RXQ9lGuyIN-9LhBKOJapZK8PQdioTVFA,8647
-nextrec/models/ranking/eulernet.py,sha256=sYOReqVY7P4s-3GwRObFn4U6WoCVubtiMk3kN6P4iKY,11276
-nextrec/models/ranking/ffm.py,sha256=xGK0K6sXQisJRhtM3q1P_hNEJ4Kletu4q4OIUcFFAZk,10304
-nextrec/models/ranking/fibinet.py,sha256=URjm2oHIE5vx_wEriZCfkaNUuJ6YTLEVG4U9uYieMbA,7075
-nextrec/models/ranking/fm.py,sha256=FJ4JbhIEYZk7UtttiOhUrCsJDcjzJTN8-L4qCv9ch4E,3587
-nextrec/models/ranking/lr.py,sha256=bfAcXuw1HGbPyDVnGK8hqe0kNpqSz4xdr-oSGXFus1E,3019
-nextrec/models/ranking/masknet.py,sha256=EWdiErRvAe4kFG-U0i2Tz3Yh_Vf-nXZNVbm6KwDYtwY,11398
-nextrec/models/ranking/pnn.py,sha256=iW3LgAKqZ1Rqzbl-8Jte9wgVyYIMzi76nDwTW78DkMU,7340
-nextrec/models/ranking/widedeep.py,sha256=EuBHnczCIT0OGxnuYEAB-5RHB_lC3Ow6xIMj9CB8o5k,4006
-nextrec/models/ranking/xdeepfm.py,sha256=MOpVsS5CQMV9scBD_WJlaBHkl2ZViq1Mjjk6Glciz4w,7247
+nextrec/models/ranking/afm.py,sha256=lLcsWB1wOaiVJEn9rtNu2EOqXCP0LIPij7d6Jo9r8KM,9331
+nextrec/models/ranking/autoint.py,sha256=vY4Lp3a5007hzXp-38lmdlLcbcI4ImVELGMV7SuWzuM,7193
+nextrec/models/ranking/dcn.py,sha256=C6sYVCExCzhuInmNaTAyUTABcx1zYi_DCE0QoWmG3Zg,6613
+nextrec/models/ranking/dcn_v2.py,sha256=5nT8rP3_0g7Uy-Kaasw_xQDu6bshOiZBCjO3SKqjKjE,10316
+nextrec/models/ranking/deepfm.py,sha256=AhP20JPT1EAa6_8mw8IbJj6OoD7Epu74iFTAEvU20LM,4338
+nextrec/models/ranking/dien.py,sha256=SqgJqSNuPDiREEzMFYyqQcoWBowJj5M4Q16BGt-aHXc,18295
+nextrec/models/ranking/din.py,sha256=fMkQ3ZzROuafbd8UY4IBst3OU_5FF3mSJMmuOPh-6ko,8789
+nextrec/models/ranking/eulernet.py,sha256=aiDBmx2K90r2Zkg5Wqc8SaK1oIAhKaFG3lJ2hkbJTCs,11418
+nextrec/models/ranking/ffm.py,sha256=a4OaauKltUUF9rUV-B1qyG54r3wYN2oRT_r-ljEyLSM,10446
+nextrec/models/ranking/fibinet.py,sha256=fMe1PSIghyt7ccca0V-o0L_vFAx62S50na6VntmCns0,7217
+nextrec/models/ranking/fm.py,sha256=qwQ_QdBn5PTMYhCDGict8JU5kdVLbk_6ZLhg1H1IdRg,3729
+nextrec/models/ranking/lr.py,sha256=ZpWbM9aMn3tzWlbe8RMmhVz0QBnIeTs5rVDINc0E11I,3161
+nextrec/models/ranking/masknet.py,sha256=h5fE4iG-YLEEpDl9gpOcJgvUsAWQGvXexojyLsrURTQ,11540
+nextrec/models/ranking/pnn.py,sha256=fFcOo4QQjA-dB_yEpOLAtFG4ZrkwpBfFo_FrPry3V9s,7482
+nextrec/models/ranking/widedeep.py,sha256=bW53VPVt7xz3c16pO2JXArp346zEQbG3NAGNqZga8G4,4201
+nextrec/models/ranking/xdeepfm.py,sha256=1H230whaeZ_uqgZsTE-jIrEmwTQPKv_7DnAcYJ6vZvg,7441
 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
@@ -83,8 +83,8 @@ nextrec/utils/loss.py,sha256=GBWQGpDaYkMJySpdG078XbeUNXUC34PVqFy0AqNS9N0,4578
 nextrec/utils/model.py,sha256=PI9y8oWz1lhktgapZsiXb8rTr2NrFFlc80tr4yOFHik,5334
 nextrec/utils/torch_utils.py,sha256=UQpWS7F3nITYqvx2KRBaQJc9oTowRkIvowhuQLt6NFM,11953
 nextrec/utils/types.py,sha256=VhtLXUVvu0zAZVAUgRUML4FExRC-GH-ZmC1UiVSr3HE,1523
-nextrec-0.4.29.dist-info/METADATA,sha256=07uY-tjvX77s4-doAM2zh5nCeptFxbzvBAv6BQdMvLU,23188
-nextrec-0.4.29.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-nextrec-0.4.29.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
-nextrec-0.4.29.dist-info/licenses/LICENSE,sha256=2fQfVKeafywkni7MYHyClC6RGGC3laLTXCNBx-ubtp0,1064
-nextrec-0.4.29.dist-info/RECORD,,
+nextrec-0.4.31.dist-info/METADATA,sha256=Pzyzo1FUubvkQEQno1BSaTRWR3NPuV5jGSuketmg0oc,23188
+nextrec-0.4.31.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+nextrec-0.4.31.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
+nextrec-0.4.31.dist-info/licenses/LICENSE,sha256=2fQfVKeafywkni7MYHyClC6RGGC3laLTXCNBx-ubtp0,1064
+nextrec-0.4.31.dist-info/RECORD,,