nextrec 0.3.6__py3-none-any.whl → 0.4.2__py3-none-any.whl

nextrec/__init__.py

@@ -1,7 +1,7 @@
 from nextrec.__version__ import __version__
 
 __all__ = [
-    '__version__',
+    "__version__",
 ]
 
 # Package metadata

nextrec/__version__.py

@@ -1 +1 @@
-__version__ = "0.3.6"
+__version__ = "0.4.2"

nextrec/basic/activation.py

@@ -14,20 +14,21 @@ class Dice(nn.Module):
     """
     Dice activation function from the paper:
     "Deep Interest Network for Click-Through Rate Prediction" (Zhou et al., 2018)
-    
+
     Dice(x) = p(x) * x + (1 - p(x)) * alpha * x
     where p(x) = sigmoid((x - E[x]) / sqrt(Var[x] + epsilon))
     """
+
     def __init__(self, emb_size: int, epsilon: float = 1e-9):
         super(Dice, self).__init__()
         self.epsilon = epsilon
         self.alpha = nn.Parameter(torch.zeros(emb_size))
         self.bn = nn.BatchNorm1d(emb_size)
-    
+
     def forward(self, x):
         # x shape: (batch_size, emb_size) or (batch_size, seq_len, emb_size)
         original_shape = x.shape
-        
+
         if x.dim() == 3:
             # For 3D input (batch_size, seq_len, emb_size), reshape to 2D
             batch_size, seq_len, emb_size = x.shape
@@ -45,7 +46,9 @@ def activation_layer(activation: str, emb_size: int | None = None):
     activation = activation.lower()
     if activation == "dice":
         if emb_size is None:
-            raise ValueError("[ActivationLayer Error]: emb_size is required for Dice activation")
+            raise ValueError(
+                "[ActivationLayer Error]: emb_size is required for Dice activation"
+            )
         return Dice(emb_size)
     elif activation == "relu":
         return nn.ReLU()
@@ -84,4 +87,6 @@ def activation_layer(activation: str, emb_size: int | None = None):
     elif activation in ["none", "linear", "identity"]:
         return nn.Identity()
     else:
-        raise ValueError(f"[ActivationLayer Error]: Unsupported activation function: {activation}")  
+        raise ValueError(
+            f"[ActivationLayer Error]: Unsupported activation function: {activation}"
+        )

nextrec/basic/callback.py

@@ -7,6 +7,7 @@ Author: Yang Zhou, zyaztec@gmail.com
 
 import copy
 
+
 class EarlyStopper(object):
     def __init__(self, patience: int = 20, mode: str = "max"):
         self.patience = patience

nextrec/basic/features.py

@@ -5,28 +5,31 @@ Date: create on 27/10/2025
 Checkpoint: edit on 02/12/2025
 Author: Yang Zhou, zyaztec@gmail.com
 """
+
 import torch
 from nextrec.utils.embedding import get_auto_embedding_dim
 from nextrec.utils.feature import normalize_to_list
 
+
 class BaseFeature(object):
     def __repr__(self):
-        params = {k: v for k, v in self.__dict__.items() if not k.startswith("_") }
+        params = {k: v for k, v in self.__dict__.items() if not k.startswith("_")}
         param_str = ", ".join(f"{k}={v!r}" for k, v in params.items())
         return f"{self.__class__.__name__}({param_str})"
 
+
 class SequenceFeature(BaseFeature):
     def __init__(
         self,
         name: str,
         vocab_size: int,
         max_len: int = 20,
-        embedding_name: str = '',
+        embedding_name: str = "",
         embedding_dim: int | None = 4,
         combiner: str = "mean",
         padding_idx: int | None = None,
-        init_type: str='normal',
-        init_params: dict|None = None,
+        init_type: str = "normal",
+        init_params: dict | None = None,
         l1_reg: float = 0.0,
         l2_reg: float = 1e-5,
         trainable: bool = True,
@@ -44,22 +47,24 @@ class SequenceFeature(BaseFeature):
         self.l1_reg = l1_reg
         self.l2_reg = l2_reg
         self.trainable = trainable
-    
+
+
 class SparseFeature(BaseFeature):
-    def __init__(self, 
-                 name: str, 
-                 vocab_size: int, 
-                 embedding_name: str = '', 
-                 embedding_dim: int | None  = 4, 
-                 padding_idx: int | None = None,
-                 init_type: str='normal',
-                 init_params: dict|None = None,
-                 l1_reg: float = 0.0,                 
-                 l2_reg: float = 1e-5,
-                 trainable: bool = True,
-                 pretrained_weight: torch.Tensor | None = None,  
-                 freeze_pretrained: bool = False,                
-                 ):
+    def __init__(
+        self,
+        name: str,
+        vocab_size: int,
+        embedding_name: str = "",
+        embedding_dim: int | None = 4,
+        padding_idx: int | None = None,
+        init_type: str = "normal",
+        init_params: dict | None = None,
+        l1_reg: float = 0.0,
+        l2_reg: float = 1e-5,
+        trainable: bool = True,
+        pretrained_weight: torch.Tensor | None = None,
+        freeze_pretrained: bool = False,
+    ):
         self.name = name
         self.vocab_size = vocab_size
         self.embedding_name = embedding_name or name
@@ -74,12 +79,13 @@ class SparseFeature(BaseFeature):
         self.pretrained_weight = pretrained_weight
         self.freeze_pretrained = freeze_pretrained
 
+
 class DenseFeature(BaseFeature):
     def __init__(
         self,
         name: str,
         embedding_dim: int | None = 1,
-        input_dim: int  = 1,
+        input_dim: int = 1,
         use_embedding: bool = False,
     ):
         self.name = name
@@ -90,6 +96,7 @@ class DenseFeature(BaseFeature):
         else:
             self.use_embedding = use_embedding  # user decides for dim <= 1
 
+
 class FeatureSet:
     def set_all_features(
         self,
@@ -103,7 +110,9 @@ class FeatureSet:
         self.sparse_features = list(sparse_features) if sparse_features else []
         self.sequence_features = list(sequence_features) if sequence_features else []
 
-        self.all_features = self.dense_features + self.sparse_features + self.sequence_features
+        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 = normalize_to_list(target)
         self.id_columns = normalize_to_list(id_columns)
@@ -115,4 +124,3 @@ class FeatureSet:
     ) -> None:
         self.target_columns = normalize_to_list(target)
         self.id_columns = normalize_to_list(id_columns)
-