nextrec 0.5.4__py3-none-any.whl → 0.5.5__py3-none-any.whl

nextrec/__version__.py

@@ -1 +1 @@
-__version__ = "0.5.4"
+__version__ = "0.5.5"

nextrec/cli.py

@@ -137,7 +137,7 @@ def train_model(train_config_path: str) -> None:
     model_cfg = read_yaml(model_cfg_path)
 
     # Extract id_column from data config for GAUC metrics
-    id_column = data_cfg.get("id_column") or data_cfg.get("user_id_column")
+    id_column = data_cfg.get("id_column")
     id_columns = [id_column] if id_column else []
 
     log_cli_section("Data")
@@ -378,6 +378,7 @@ def train_model(train_config_path: str) -> None:
         sparse_features,
         sequence_features,
         target,
+        id_columns,
         device,
     )
 
@@ -585,14 +586,16 @@ def predict_model(predict_config_path: str) -> None:
         sparse_features=sparse_features,
         sequence_features=sequence_features,
         target=target_cols,
+        id_columns=id_columns,
         device=predict_cfg.get("device", "cpu"),
     )
-    model.id_columns = id_columns
+
     model.load_model(
         model_file, map_location=predict_cfg.get("device", "cpu"), verbose=True
     )
 
-    effective_id_columns = model.id_columns
+    effective_id_columns = id_columns or model.id_columns
+
     log_cli_section("Features")
     log_kv_lines(
         [

nextrec/utils/config.py

@@ -388,6 +388,7 @@ def build_model_instance(
     sparse_features: List["SparseFeature"],
     sequence_features: List["SequenceFeature"],
     target: List[str],
+    id_columns: list[str] | str | None,
     device: str,
 ) -> Any:
     """
@@ -400,6 +401,7 @@ def build_model_instance(
         sparse_features: List of sparse feature objects
         sequence_features: List of sequence feature objects
         target: List of target column names
+        id_columns: Identifier column name(s) for GAUC or ID passthrough
         device: Device string (e.g., 'cpu', 'cuda:0')
     """
     dense_map = {f.name: f for f in dense_features}
@@ -436,7 +438,7 @@ def build_model_instance(
         param.kind == inspect.Parameter.VAR_KEYWORD for param in sig_params.values()
     )
 
-    init_kwargs: Dict[str, Any] = dict(params_cfg)
+    init_kwargs = dict(params_cfg)
 
     # Explicit bindings (model_config.feature_bindings) take priority
     for param_name, binding in feature_bindings_cfg.items():
@@ -517,6 +519,8 @@ def build_model_instance(
 
     if accepts("target"):
         init_kwargs.setdefault("target", target)
+    if accepts("id_columns") or accepts_var_kwargs:
+        init_kwargs.setdefault("id_columns", id_columns)
     if accepts("device"):
         init_kwargs.setdefault("device", device)
 

{nextrec-0.5.4.dist-info → nextrec-0.5.5.dist-info}/METADATA

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

{nextrec-0.5.4.dist-info → nextrec-0.5.5.dist-info}/RECORD

@@ -1,6 +1,6 @@
 nextrec/__init__.py,sha256=_M3oUqyuvQ5k8Th_3wId6hQ_caclh7M5ad51XN09m98,235
-nextrec/__version__.py,sha256=DITpct-LrdIsTgwx2NgH5Ghx5y8Xgz1YMimy1ZV5RTY,22
-nextrec/cli.py,sha256=IxYzZavCc5ACe-8i2EJbkdY66do4x7siGHSxoS1RoL0,29676
+nextrec/__version__.py,sha256=78mfpLewKVki6c9UONSUdlVme_JsN9ZwIfp4Hf4jmG0,22
+nextrec/cli.py,sha256=ywGOMJ-iVLH6bOk2VX_n0r530V11xbFUHVwu6-AlzPw,29675
 nextrec/basic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nextrec/basic/activation.py,sha256=rU-W-DHgiD3AZnMGmD014ChxklfP9BpedDTiwtdgXhA,2762
 nextrec/basic/asserts.py,sha256=eaB4FZJ7Sbh9S8PLJZNiYd7Zi98ca2rDi4S7wSYCaEw,1473
@@ -71,7 +71,7 @@ nextrec/models/tree_base/catboost.py,sha256=hXINyx7iianwDxOZx3SLm0i-YP1jiC3HcAeq
 nextrec/models/tree_base/lightgbm.py,sha256=VilMU7SgfHR5LAaaoQo-tY1vkzpSvWovIrgaSeuJ1-A,2263
 nextrec/models/tree_base/xgboost.py,sha256=thOmDIC_nitno_k2mcH2cj2VcS07f9veTG01FMOO-28,1957
 nextrec/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/utils/config.py,sha256=Ngd_u8ZS5br4lIqrBJ_ecLquMF4KJi6TPAGqLZg8H4s,20485
+nextrec/utils/config.py,sha256=uChqdxA3X6gn2G8tlUrobZ1AbaxdSQMt87eocSQVgog,20691
 nextrec/utils/console.py,sha256=RnSUplJnyanSQ6TyMQkP7S1j2rGMver1DbFVqNH6_1k,13581
 nextrec/utils/data.py,sha256=JN30npycFMo_caBQJ5815tciRW3r7YDqSQxnJuXhlVA,22299
 nextrec/utils/embedding.py,sha256=akAEc062MG2cD7VIOllHaqtwzAirQR2gq5iW7oKpGAU,1449
@@ -80,8 +80,8 @@ nextrec/utils/model.py,sha256=K1-ZfS1umSTVo-lNHKNkBb3xvWnfiJJSTRicSVDpk4s,5148
 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.5.4.dist-info/METADATA,sha256=OlEXZ7qVUN_3IsAFoDxmERQYxGro70BDo3v3dy_8zec,23464
-nextrec-0.5.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-nextrec-0.5.4.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
-nextrec-0.5.4.dist-info/licenses/LICENSE,sha256=COP1BsqnEUwdx6GCkMjxOo5v3pUe4-Go_CdmQmSfYXM,1064
-nextrec-0.5.4.dist-info/RECORD,,
+nextrec-0.5.5.dist-info/METADATA,sha256=XbDqdm6X1jgRyk_GSMXyMVKtZe1AsuCNTNjcC4NTD-A,23464
+nextrec-0.5.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+nextrec-0.5.5.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
+nextrec-0.5.5.dist-info/licenses/LICENSE,sha256=COP1BsqnEUwdx6GCkMjxOo5v3pUe4-Go_CdmQmSfYXM,1064
+nextrec-0.5.5.dist-info/RECORD,,