nextrec 0.4.18__py3-none-any.whl → 0.4.20__py3-none-any.whl

nextrec/__version__.py

@@ -1 +1 @@
-__version__ = "0.4.18"
+__version__ = "0.4.20"

nextrec/cli.py

@@ -85,7 +85,8 @@ def train_model(train_config_path: str) -> None:
     session_id = session_cfg.get("id", "nextrec_cli_session")
     artifact_root = Path(session_cfg.get("artifact_root", "nextrec_logs"))
     session_dir = artifact_root / session_id
-    setup_logger(session_id=session_id)
+
+    setup_logger(session_id=session_dir.resolve())
 
     log_cli_section("CLI")
     log_kv_lines(
@@ -419,8 +420,7 @@ def predict_model(predict_config_path: str) -> None:
         session_cfg = cfg.get("session", {}) or {}
         session_id = session_cfg.get("id") or session_dir.name
 
-    setup_logger(session_id=session_id)
-
+    setup_logger(session_id=session_dir.resolve())
     log_cli_section("CLI")
     log_kv_lines(
         [
@@ -585,13 +585,12 @@ def predict_model(predict_config_path: str) -> None:
         chunk_size=predict_cfg.get("chunk_size", 20000),
     )
 
-    # Build output path: {checkpoint_path}/predictions/{name}.{save_data_format}
     save_format = predict_cfg.get(
         "save_data_format", predict_cfg.get("save_format", "csv")
     )
     pred_name = predict_cfg.get("name", "pred")
-    # Pass filename with extension to let model.predict handle path resolution
-    save_path = f"{pred_name}.{save_format}"
+
+    save_path = checkpoint_base / "predictions" / f"{pred_name}.{save_format}"
 
     start = time.time()
     logger.info("")
@@ -600,7 +599,7 @@ def predict_model(predict_config_path: str) -> None:
         batch_size=batch_size,
         include_ids=bool(id_columns),
         return_dataframe=False,
-        save_path=save_path,
+        save_path=str(save_path),
         save_format=save_format,
         num_workers=predict_cfg.get("num_workers", 0),
     )

{nextrec-0.4.18.dist-info → nextrec-0.4.20.dist-info}/METADATA

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

{nextrec-0.4.18.dist-info → nextrec-0.4.20.dist-info}/RECORD

@@ -1,6 +1,6 @@
 nextrec/__init__.py,sha256=_M3oUqyuvQ5k8Th_3wId6hQ_caclh7M5ad51XN09m98,235
-nextrec/__version__.py,sha256=my_eRKQWrGqOjGcYUQ3l4O21n-LMG22J283Pb5L90kw,23
-nextrec/cli.py,sha256=xeVkW1rMLExu8AZ66jqOe4dNIh22pNz2Cbj-dtpG0_A,24445
+nextrec/__version__.py,sha256=WCSj_sB9bdEmI7lwg4E_Vhy7TRLJmGYIYjp5bfMMYXE,23
+nextrec/cli.py,sha256=rHATyy1v7ERPqkHNUf4-JTlRMdmlfgHd6VBNKR-Ik7o,24347
 nextrec/basic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nextrec/basic/activation.py,sha256=uzTWfCOtBSkbu_Gk9XBNTj8__s241CaYLJk6l8nGX9I,2885
 nextrec/basic/callback.py,sha256=nn1f8FG9c52vJ-gvwteqPbk3-1QuNS1vmhBlkENdb0I,14636
@@ -72,8 +72,8 @@ nextrec/utils/embedding.py,sha256=akAEc062MG2cD7VIOllHaqtwzAirQR2gq5iW7oKpGAU,14
 nextrec/utils/feature.py,sha256=rsUAv3ELyDpehVw8nPEEsLCCIjuKGTJJZuFaWB_wrPk,633
 nextrec/utils/model.py,sha256=fHvFciUuMOVcM1oWiRva4LcArRdZ1R5Uzml-COSqqvM,4688
 nextrec/utils/torch_utils.py,sha256=AKfYbSOJjEw874xsDB5IO3Ote4X7vnqzt_E0jJny0o8,13468
-nextrec-0.4.18.dist-info/METADATA,sha256=5UMrAXqxBzwFPR1SwEMh_9sSOnUA9FnUQj1hF9PTt5w,21301
-nextrec-0.4.18.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-nextrec-0.4.18.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
-nextrec-0.4.18.dist-info/licenses/LICENSE,sha256=2fQfVKeafywkni7MYHyClC6RGGC3laLTXCNBx-ubtp0,1064
-nextrec-0.4.18.dist-info/RECORD,,
+nextrec-0.4.20.dist-info/METADATA,sha256=k_ph4RolNDzWnSYvhiIjopNiQOJHPs2fgeUcyg7-h80,21482
+nextrec-0.4.20.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+nextrec-0.4.20.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
+nextrec-0.4.20.dist-info/licenses/LICENSE,sha256=2fQfVKeafywkni7MYHyClC6RGGC3laLTXCNBx-ubtp0,1064
+nextrec-0.4.20.dist-info/RECORD,,