nextrec 0.4.30__py3-none-any.whl → 0.4.32__py3-none-any.whl

nextrec/models/tree_base/catboost.py

@@ -0,0 +1,97 @@
+"""
+CatBoost adapter for NextRec.
+"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import numpy as np
+import pandas as pd
+
+from nextrec.models.tree_base.base import TreeBaseModel
+
+
+class Catboost(TreeBaseModel):
+    model_file_suffix = "cbm"
+
+    @property
+    def model_name(self) -> str:
+        return "catboost"
+
+    def build_estimator(self, model_params: dict[str, Any], epochs: int | None):
+        try:
+            import catboost as cb
+        except ModuleNotFoundError as exc:  # pragma: no cover
+            raise ModuleNotFoundError(
+                "catboost is required for model='catboost'. Install with: pip install catboost"
+            ) from exc
+        params = dict(model_params)
+        if epochs is not None and "iterations" not in params:
+            params["iterations"] = epochs
+        task = self.task[0] if isinstance(self.task, list) else self.task
+        if task == "regression":
+            params.setdefault("loss_function", "RMSE")
+            return cb.CatBoostRegressor(**params)
+        if task == "binary":
+            params.setdefault("loss_function", "Logloss")
+            return cb.CatBoostClassifier(**params)
+        raise ValueError(f"[catboost-init Error] Unsupported task type: {task}")
+
+    def fit_estimator(
+        self,
+        model: Any,
+        X_train,
+        y_train,
+        X_valid,
+        y_valid,
+        cat_features,
+        **kwargs,
+    ):
+        fit_kwargs = dict(kwargs)
+        X_train, cat_features = self._prepare_catboost_data(X_train, cat_features)
+        if cat_features:
+            fit_kwargs.setdefault("cat_features", cat_features)
+        if X_valid is not None and y_valid is not None:
+            X_valid, _ = self._prepare_catboost_data(X_valid, cat_features)
+            fit_kwargs.setdefault("eval_set", (X_valid, y_valid))
+        model.fit(X_train, y_train, **fit_kwargs)
+        return model
+
+    def save_model_file(self, model: Any, path) -> None:
+        model.save_model(str(path))
+
+    def load_model_file(self, path):
+        model = self.build_estimator(dict(self.model_params), epochs=None)
+        model.load_model(str(path))
+        return model
+
+    def _prepare_catboost_data(
+        self, X: Any, cat_features: list[int]
+    ) -> tuple[Any, list[int]]:
+        if not cat_features:
+            return X, cat_features
+        if isinstance(X, np.ndarray) and np.issubdtype(X.dtype, np.floating):
+            feature_names = [f"f{idx}" for idx in range(X.shape[1])]
+            df = pd.DataFrame(X, columns=feature_names)
+            for idx in cat_features:
+                if 0 <= idx < df.shape[1]:
+                    series = df.iloc[:, idx]
+                    df.iloc[:, idx] = series.map(
+                        lambda value: (
+                            ""
+                            if pd.isna(value)
+                            else (
+                                str(int(value))
+                                if isinstance(value, (float, np.floating))
+                                and value.is_integer()
+                                else str(value)
+                            )
+                        )
+                    )
+            return df, cat_features
+        return X, cat_features
+
+    def predict_scores(self, model: Any, X: np.ndarray) -> np.ndarray:
+        X, _ = self._prepare_catboost_data(X, self._cat_feature_indices)
+        return super().predict_scores(model, X)

nextrec/models/tree_base/lightgbm.py

@@ -0,0 +1,69 @@
+"""
+LightGBM adapter for NextRec.
+"""
+
+from __future__ import annotations
+
+from typing import Any
+
+from nextrec.models.tree_base.base import TreeBaseModel
+
+
+class Lightgbm(TreeBaseModel):
+    model_file_suffix = "txt"
+
+    @property
+    def model_name(self) -> str:
+        return "lightgbm"
+
+    def build_estimator(self, model_params: dict[str, Any], epochs: int | None):
+        try:
+            import lightgbm as lgb
+        except ModuleNotFoundError as exc:  # pragma: no cover
+            raise ModuleNotFoundError(
+                "lightgbm is required for model='lightgbm'. Install with: pip install lightgbm"
+            ) from exc
+        params = dict(model_params)
+        if epochs is not None and "n_estimators" not in params:
+            params["n_estimators"] = epochs
+        task = self.task[0] if isinstance(self.task, list) else self.task
+        if task == "regression":
+            params.setdefault("objective", "regression")
+            return lgb.LGBMRegressor(**params)
+        if task == "binary":
+            params.setdefault("objective", "binary")
+            return lgb.LGBMClassifier(**params)
+        raise ValueError(f"[lightgbm-init Error] Unsupported task type: {task}")
+
+    def fit_estimator(
+        self,
+        model: Any,
+        X_train,
+        y_train,
+        X_valid,
+        y_valid,
+        cat_features,
+        **kwargs,
+    ):
+        fit_kwargs = dict(kwargs)
+        if cat_features:
+            fit_kwargs.setdefault("categorical_feature", cat_features)
+        if X_valid is not None and y_valid is not None:
+            fit_kwargs.setdefault("eval_set", [(X_valid, y_valid)])
+        model.fit(X_train, y_train, **fit_kwargs)
+        return model
+
+    def save_model_file(self, model: Any, path) -> None:
+        if hasattr(model, "booster_"):
+            model.booster_.save_model(str(path))
+        else:
+            model.save_model(str(path))
+
+    def load_model_file(self, path):
+        try:
+            import lightgbm as lgb
+        except ModuleNotFoundError as exc:  # pragma: no cover
+            raise ModuleNotFoundError(
+                "lightgbm is required for model='lightgbm'. Install with: pip install lightgbm"
+            ) from exc
+        return lgb.Booster(model_file=str(path))

nextrec/models/tree_base/xgboost.py

@@ -0,0 +1,61 @@
+"""
+XGBoost adapter for NextRec.
+"""
+
+from __future__ import annotations
+
+from typing import Any
+
+from nextrec.models.tree_base.base import TreeBaseModel
+
+
+class Xgboost(TreeBaseModel):
+    model_file_suffix = "json"
+
+    @property
+    def model_name(self) -> str:
+        return "xgboost"
+
+    def build_estimator(self, model_params: dict[str, Any], epochs: int | None):
+        try:
+            import xgboost as xgb
+        except ModuleNotFoundError as exc:  # pragma: no cover
+            raise ModuleNotFoundError(
+                "xgboost is required for model='xgboost'. Install with: pip install xgboost"
+            ) from exc
+        params = dict(model_params)
+        if epochs is not None and "n_estimators" not in params:
+            params["n_estimators"] = epochs
+        task = self.task[0] if isinstance(self.task, list) else self.task
+        if task == "regression":
+            params.setdefault("objective", "reg:squarederror")
+            return xgb.XGBRegressor(**params)
+        if task == "binary":
+            params.setdefault("objective", "binary:logistic")
+            return xgb.XGBClassifier(**params)
+        raise ValueError(f"[xgboost-init Error] Unsupported task type: {task}")
+
+    def fit_estimator(
+        self,
+        model: Any,
+        X_train,
+        y_train,
+        X_valid,
+        y_valid,
+        cat_features,
+        **kwargs,
+    ):
+        del cat_features  # xgboost handles categorical features via params
+        fit_kwargs = dict(kwargs)
+        if X_valid is not None and y_valid is not None:
+            fit_kwargs.setdefault("eval_set", [(X_valid, y_valid)])
+        model.fit(X_train, y_train, **fit_kwargs)
+        return model
+
+    def save_model_file(self, model: Any, path) -> None:
+        model.save_model(str(path))
+
+    def load_model_file(self, path):
+        model = self.build_estimator(dict(self.model_params), epochs=None)
+        model.load_model(str(path))
+        return model

nextrec/utils/config.py

@@ -342,6 +342,7 @@ def load_model_class(model_cfg: Dict[str, Any], base_dir: Path) -> type:
             f"nextrec.models.match.{name.lower()}",
             f"nextrec.models.multi_task.{name.lower()}",
             f"nextrec.models.generative.{name.lower()}",
+            f"nextrec.models.tree_base.{name.lower()}",
         ]
         errors = []
 

nextrec/utils/types.py

@@ -62,6 +62,8 @@ TrainingModeName = Literal["pointwise", "pairwise", "listwise"]
 
 TaskTypeName = Literal["binary", "regression"]
 
+TaskTypeInput = TaskTypeName | str
+
 MetricsName = Literal[
     "auc",
     "gauc",

{nextrec-0.4.30.dist-info → nextrec-0.4.32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextrec
-Version: 0.4.30
+Version: 0.4.32
 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.30-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.4.32-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.30，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
+> 截止当前版本0.4.32，NextRec CLI支持单机训练，分布式训练相关功能尚在开发中。
 
 ## 兼容平台
 
-当前最新版本为0.4.30，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
+当前最新版本为0.4.32，所有模型和测试代码均已在以下平台通过验证，如果开发者在使用中遇到兼容问题，请在issue区提出错误报告及系统版本：
 
 | 平台 | 配置 | 
 |------|------|
@@ -400,7 +400,7 @@ NextRec 的开发受到以下优秀项目的启发：
 @misc{nextrec,
     title = {NextRec},
     author = {Yang Zhou},
-    year = {2025},
+    year = {2026},
     publisher = {GitHub},
     journal = {GitHub repository},
     howpublished = {\url{https://github.com/zerolovesea/NextRec}},

{nextrec-0.4.30.dist-info → nextrec-0.4.32.dist-info}/RECORD

@@ -1,6 +1,6 @@
 nextrec/__init__.py,sha256=_M3oUqyuvQ5k8Th_3wId6hQ_caclh7M5ad51XN09m98,235
-nextrec/__version__.py,sha256=oJL8XtG8mmnvRfpE8L0X7P-SyENhGR162XgumWN6O4A,23
-nextrec/cli.py,sha256=uOaXnlAM-ARrbxKOVWWkTE_rv-54px168kBhFUHtIAg,25073
+nextrec/__version__.py,sha256=W0DtmvTLu6FQL6tby9DrJltesCOu7Q36WFhsT2wLrgM,23
+nextrec/cli.py,sha256=hFDL_HlukJxdp4FU486g977Rix9OkGdEPGBj2HxqCGo,25393
 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
@@ -10,15 +10,15 @@ nextrec/basic/heads.py,sha256=BshykLxD41KxKuZaBxf4Fmy1Mc52b3ioJliN1BVaGlk,3374
 nextrec/basic/layers.py,sha256=tr8XFOcTvUHEZ6T3zJwmtKMA-u_xfzHloIkItGs821U,40084
 nextrec/basic/loggers.py,sha256=LAfnhdSNEzHybrXaKxCWoAML1c2A-FJF6atpfrrm_Kw,13840
 nextrec/basic/metrics.py,sha256=CPzENDcpO6QTDZLBtQlfAGKUYYQc0FT-eaMKJ4MURFo,23396
-nextrec/basic/model.py,sha256=4vBp-vXAWC5Oiu_x4mtVaXTKJCcKDYT0IJ7UOyHD5lw,110162
+nextrec/basic/model.py,sha256=uAC3wFKJcRUAgsvfc9hXhhfp1iILqvTSbA7Ryohn-bc,111590
 nextrec/basic/session.py,sha256=mrIsjRJhmvcAfoO1pXX-KB3SK5CCgz89wH8XDoAiGEI,4475
 nextrec/basic/summary.py,sha256=b6jLo70gqZj_bQ4eb5yb8SXmr2ilZlKNN293EyVnkyc,17759
 nextrec/data/__init__.py,sha256=YZQjpty1pDCM7q_YNmiA2sa5kbujUw26ObLHWjMPjKY,1194
-nextrec/data/batch_utils.py,sha256=vqlBvdfrhFEIUHmiWuUdXg1oY2jkyKOO1JO2l2KpJUU,3549
+nextrec/data/batch_utils.py,sha256=TbnXYqYlmK51dJthaL6dO7LTn4wyp8221I-kdgvpvDE,3542
 nextrec/data/data_processing.py,sha256=lhuwYxWp4Ts2bbuLGDt2LmuPrOy7pNcKczd2uVcQ4ss,6476
 nextrec/data/data_utils.py,sha256=0Ls1cnG9lBz0ovtyedw5vwp7WegGK_iF-F8e_3DEddo,880
 nextrec/data/dataloader.py,sha256=2sXwoiWxupKE-V1QYeZlXjK1yJyxhDtlOhknAnJF8Wk,19727
-nextrec/data/preprocessor.py,sha256=AD5bHNbkAZAnI_SbDfJJaAh57CRtRjoOQJ6aIBkgoQs,65251
+nextrec/data/preprocessor.py,sha256=n2ZDR4o_-5nouBgCluWlVrXRkA9AoRaO7EvXPZAQvJg,66734
 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,37 +26,37 @@ nextrec/loss/pairwise.py,sha256=9fyH9p2u-N0-jAnNTq3X5Dje0ipj1dob8wp-yQKRra4,3493
 nextrec/loss/pointwise.py,sha256=09nzI1L5eP9raXnj3Q49bD9Clp_JmsSWUvEj7bkTzSw,7474
 nextrec/models/generative/__init__.py,sha256=0MV3P-_ainPaTxmRBGWKUVCEt14KJvuvEHmRB3OQ1Fs,176
 nextrec/models/generative/tiger.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/models/multi_task/[pre]aitm.py,sha256=e9XCuPgOoyTs1RlGSJsMMksyYFOrOaT00c4WW4TSg2g,6659
-nextrec/models/multi_task/[pre]snr_trans.py,sha256=Mb1RSdBAVCKhQnG9ajf1uYHulZsK8rz5sLGWpTJVMwY,9060
-nextrec/models/multi_task/[pre]star.py,sha256=hJ-E0_ciiOlaOhQ-gh3UZpZcKLc8A6_7CiVJyurLy_E,7040
+nextrec/models/multi_task/[pre]aitm.py,sha256=A2n0T4JEui-uHgbqITU5lpsmtnP14fQXRZM1peTPvhQ,6661
+nextrec/models/multi_task/[pre]snr_trans.py,sha256=k08tC-TI--a_Tt4_BmX0ZubzntyqwsejutYzbB5F4S4,9077
+nextrec/models/multi_task/[pre]star.py,sha256=BczXHPJtK7xyPbLO0fQ-w7qnzaBvLpyhG0CKMBUItCY,7057
 nextrec/models/multi_task/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nextrec/models/multi_task/apg.py,sha256=oqX3-wShX47Zqn6o0WGJ5duqjnxBKuMd1XodwsxVDiQ,13519
-nextrec/models/multi_task/cross_stitch.py,sha256=jMYd-BVhkPtJRgpzVPooXCzf_LzS0oWuGKanNUlytBo,9424
-nextrec/models/multi_task/escm.py,sha256=Djk4SjLsZ8He3sZrKfZkHdH2istR9R71GVEJc6EbF-o,11021
-nextrec/models/multi_task/esmm.py,sha256=QRnNXV1IEArYorYJLOoBcLbxarMC_7azQLP79QTfAJ8,5273
-nextrec/models/multi_task/hmoe.py,sha256=6mTzZxC5PfSQovrmnR0O2hdhDUG_8yNqMwCdkNRlHkY,7567
-nextrec/models/multi_task/mmoe.py,sha256=Mplzstu-LYEMtiKiyiMEQZwY-xMkUeG-1FaDqqK5kws,7606
-nextrec/models/multi_task/pepnet.py,sha256=ikCDu66TTPdfp94HU-4QzKCW5JOEvwAf5E0QEQ0SZzw,13510
-nextrec/models/multi_task/ple.py,sha256=cO-NqEm-UZKRz2MznBjqsXL8ImH7WU1HRzXdWAtb7Kk,12089
-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/multi_task/apg.py,sha256=dNveyottpHTd811_3KVZQQqgffDzmX_rY1EMiv1oKeo,13536
+nextrec/models/multi_task/cross_stitch.py,sha256=o48ZZFKWXz-w4MWboVdxDxifM0V3_n2Is2nZ2HoJkfw,9441
+nextrec/models/multi_task/escm.py,sha256=9kc50zzFm3HJMhQDqt6kAAJQTt0hKM5luLSC9G4N7Vk,11038
+nextrec/models/multi_task/esmm.py,sha256=iTRMcjCznyKlwJMGWgqXl9CkQrvGTsfjEjpU_5X12Kg,5275
+nextrec/models/multi_task/hmoe.py,sha256=Ge0AC7oLcEwvt75AwGG5MtClCX6L46H7HRDqD9h-kpQ,7584
+nextrec/models/multi_task/mmoe.py,sha256=vc19O4N4_64-_oucOGI-P_LaxsQaGvrYGg5Z2sxP49w,7609
+nextrec/models/multi_task/pepnet.py,sha256=I0MXLXAKBRTN1Vp1QOZD47IrljDDB3UOPHZYUH1cghU,13526
+nextrec/models/multi_task/ple.py,sha256=tr8mendvt87u7a4P54lQBYCHtsWDae7VrN9fOqHcXXo,12154
+nextrec/models/multi_task/poso.py,sha256=jr-RaLl5UnZc1HcEIK3HrNnc_g17BImyJb43d4rEXpE,18218
+nextrec/models/multi_task/share_bottom.py,sha256=NYK2B9k7BaXxzX4VN3dE0_quwsOQGM7JElm2tBdb9MY,5237
 nextrec/models/ranking/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-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/ranking/afm.py,sha256=2IylgCi7DBZzF7bdWFWvsEBlnRmyradxdyJiz5P4Nwg,9397
+nextrec/models/ranking/autoint.py,sha256=5g0tMtvkt3FiCjqmU76K7w7z-h3nMJbP0eVt5Ph4bbU,7259
+nextrec/models/ranking/dcn.py,sha256=qG3O2oL9ZLu9NBOJST06JeEh0k6UTXs-X1mQe_H4QCE,6679
+nextrec/models/ranking/dcn_v2.py,sha256=807wASeG56WYD7mEDEKaERw3r-Jpas6WhDzO0HGEk9I,10382
+nextrec/models/ranking/deepfm.py,sha256=1fuc-9f8DKDUH7EAY-XX5lITJ7Qw7ge6PTWFfQ7wld8,4404
+nextrec/models/ranking/dien.py,sha256=cluW44zOBHadf3oup6h7MpTexDXjH_VPkYcqsNBT80Y,18361
+nextrec/models/ranking/din.py,sha256=J6-S72_KJYLrzUmdrh6aAx-Qc1C08ZY7lY_KFtAkJz0,8855
+nextrec/models/ranking/eulernet.py,sha256=0nOBfccfvukSZLUNOCcB_azCh52DGJq-s9doyEGMN8E,11484
+nextrec/models/ranking/ffm.py,sha256=v15x2-rExcrEYdcPf2IxEgx-ImDSevhkhi4Oe4GbloY,10512
+nextrec/models/ranking/fibinet.py,sha256=ejR1vNh5XM23SD7mfT686kuv3cmf5gKfkj0z_iMQqNA,7283
+nextrec/models/ranking/fm.py,sha256=SlFtbtnrZbeRnCHf-kUAMaeLV_wDgLiaBwPGeAO_ycM,3795
+nextrec/models/ranking/lr.py,sha256=0gmqPED-z7k4WVRy11WSLhYfS4bJQPRQTzQbe-rUITg,3227
+nextrec/models/ranking/masknet.py,sha256=brRh8r-3TqJ9aNGj2JggAB9kUfD2xff6UmTYNd96YU4,11606
+nextrec/models/ranking/pnn.py,sha256=NU537ySMMXtndtPuiwCCQhZhBpGk6X6PRDa55GFdqt0,7548
+nextrec/models/ranking/widedeep.py,sha256=i37lqEUZUtxNXSkkhQjBF9QP6yreGdV_jcphofHOzW4,4267
+nextrec/models/ranking/xdeepfm.py,sha256=vqDZRlrY9tgoqvnTXWe8xUbFx_28VztgVVc6ukzevRc,7507
 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
@@ -73,8 +73,13 @@ nextrec/models/retrieval/sdm.py,sha256=1Y2gidG7WKuuGFaaQ8BcBGhQYoyyLPyhpRTo_xE1p
 nextrec/models/retrieval/youtube_dnn.py,sha256=hLyR4liuusJIjRg4vuaSoSEecYgDICipXnNFiA3o3oY,6351
 nextrec/models/sequential/hstu.py,sha256=iZcYLp44r23nHYNhGwD25JfH85DBrFwHOTg1WpHvLe8,18983
 nextrec/models/sequential/sasrec.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nextrec/models/tree_base/__init__.py,sha256=ssGpU1viVidr1tgKCyvmkUXe32bUmwb5qPEkTh7ce70,342
+nextrec/models/tree_base/base.py,sha256=6aXYM_nA1XkKfo3KfWSYBoO6IqPIafLrdzJ334ZBaZY,26558
+nextrec/models/tree_base/catboost.py,sha256=hXINyx7iianwDxOZx3SLm0i-YP1jiC3HcAeqP9A2i4A,3434
+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=Td-TC1IoTeb0KV-EgPy_vTHmmxmE6tO9q7Gmgsk1p-A,2672
-nextrec/utils/config.py,sha256=WeGHjoQYA5SoC9B_uS3D6ChzKr9Z5n2fwE-8l6nsuLE,20425
+nextrec/utils/config.py,sha256=SUKVgWrsCkJvKLBwcHQHls859jhdPzXk7_3DYoyIXzE,20481
 nextrec/utils/console.py,sha256=RA3ZTjtUQXvueouSmXJNLkRjeUGQZesphwWjFMTbV4I,13577
 nextrec/utils/data.py,sha256=pSL96mWjWfW_RKE-qlUSs9vfiYnFZAaRirzA6r7DB6s,24994
 nextrec/utils/embedding.py,sha256=akAEc062MG2cD7VIOllHaqtwzAirQR2gq5iW7oKpGAU,1449
@@ -82,9 +87,9 @@ nextrec/utils/feature.py,sha256=E3NOFIW8gAoRXVrDhCSonzg8k7nMUZyZzMfCq9k73_A,623
 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.30.dist-info/METADATA,sha256=Bv8blGCAsM6dGbFTa_QYpH3TmbfWENF3JgH4H83ON2Q,23188
-nextrec-0.4.30.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-nextrec-0.4.30.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
-nextrec-0.4.30.dist-info/licenses/LICENSE,sha256=2fQfVKeafywkni7MYHyClC6RGGC3laLTXCNBx-ubtp0,1064
-nextrec-0.4.30.dist-info/RECORD,,
+nextrec/utils/types.py,sha256=G88DHXFv-mbg-XY-7Xwwh1qvh6WM9jpAsBjw5VuBcio,1559
+nextrec-0.4.32.dist-info/METADATA,sha256=QkHGZMQg5HZLeO0PpByGa-FiNAeclrEeLinbF_K0Jik,23188
+nextrec-0.4.32.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+nextrec-0.4.32.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
+nextrec-0.4.32.dist-info/licenses/LICENSE,sha256=COP1BsqnEUwdx6GCkMjxOo5v3pUe4-Go_CdmQmSfYXM,1064
+nextrec-0.4.32.dist-info/RECORD,,

{nextrec-0.4.30.dist-info → nextrec-0.4.32.dist-info}/licenses/LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2025 NextRec
+Copyright (c) 2026 NextRec
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal