nextrec 0.2.4__tar.gz → 0.2.6__tar.gz

{nextrec-0.2.4 → nextrec-0.2.6}/.gitignore

@@ -115,10 +115,7 @@ dmypy.json
 .temp/
 
 # test files
-local_deepfm.py
-local_mmoe.py
-online_test.py
-linux_train_tutorial.py
+*iflytek*
 
 # DataProcessor saved files
 processor/

{nextrec-0.2.4 → nextrec-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextrec
-Version: 0.2.4
+Version: 0.2.6
 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
@@ -61,7 +61,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.2.4-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.2.6-orange.svg)
 
 English | [中文版](README_zh.md)
 

{nextrec-0.2.4 → nextrec-0.2.6}/README.md

@@ -5,7 +5,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.2.4-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.2.6-orange.svg)
 
 English | [中文版](README_zh.md)
 

{nextrec-0.2.4 → nextrec-0.2.6}/README_zh.md

@@ -5,7 +5,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.2.4-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.2.6-orange.svg)
 
 [English Version](README.md) | 中文版
 

{nextrec-0.2.4 → nextrec-0.2.6}/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.2.4"
+release = "0.2.6"
 
 extensions = [
     "myst_parser",

{nextrec-0.2.4 → nextrec-0.2.6}/docs/rtd/index.md

@@ -126,8 +126,8 @@ model.fit(
 
 scores = model.evaluate(valid_df_or_loader)    # returns metric dict
 preds = model.predict(test_df_or_loader)       # numpy array or dict
-model.save_weights("checkpoint.model")
-model.load_weights("checkpoint.model", map_location="cpu")
+model.save_model("checkpoint.model")              # default no timestamp (overwrite)
+model.load_model("checkpoint.model", map_location="cpu")
 ```
 
 ## Model zoo (`nextrec.models`)

nextrec-0.2.6/nextrec/__version__.py

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

{nextrec-0.2.4 → nextrec-0.2.6}/nextrec/basic/features.py

@@ -93,6 +93,8 @@ class FeatureSpecMixin:
         dense_features: Sequence[DenseFeature] | None = None,
         sparse_features: Sequence[SparseFeature] | None = None,
         sequence_features: Sequence[SequenceFeature] | None = None,
+        target: str | Sequence[str] | None = None,
+        id_columns: str | Sequence[str] | None = None,
     ) -> None:
         self.dense_features: List[DenseFeature] = list(dense_features) if dense_features else []
         self.sparse_features: List[SparseFeature] = list(sparse_features) if sparse_features else []
@@ -100,8 +102,10 @@ class FeatureSpecMixin:
 
         self.all_features = self.dense_features + self.sparse_features + self.sequence_features
         self.feature_names = [feat.name for feat in self.all_features]
+        self.target_columns = self._normalize_to_list(target)
+        self.id_columns = self._normalize_to_list(id_columns)
 
-    def _set_target_config(
+    def _set_target_id_config(
         self,
         target: str | Sequence[str] | None = None,
         id_columns: str | Sequence[str] | None = None,

{nextrec-0.2.4 → nextrec-0.2.6}/nextrec/basic/layers.py

@@ -49,10 +49,6 @@ __all__ = [
 
 
 class PredictionLayer(nn.Module):
-    _CLASSIFICATION_TASKS = {"classification", "binary", "ctr", "ranking", "match", "matching"}
-    _REGRESSION_TASKS = {"regression", "continuous"}
-    _MULTICLASS_TASKS = {"multiclass", "softmax"}
-
     def __init__(
         self,
         task_type: Union[str, Sequence[str]] = "binary",
@@ -131,11 +127,11 @@ class PredictionLayer(nn.Module):
 
     def _get_activation(self, task_type: str):
         task = task_type.lower()
-        if task in self._CLASSIFICATION_TASKS:
+        if task in ['binary','multiclass']:
             return torch.sigmoid
-        if task in self._REGRESSION_TASKS:
+        if task in ['regression']:
             return lambda x: x
-        if task in self._MULTICLASS_TASKS:
+        if task in ['multiclass']:
             return lambda x: torch.softmax(x, dim=-1)
         raise ValueError(f"Unsupported task_type '{task_type}'.")