nextrec 0.1.3__py3-none-any.whl → 0.1.7__py3-none-any.whl

nextrec/basic/features.py

@@ -9,26 +9,28 @@ Author:
 from typing import Optional
 from nextrec.utils import get_auto_embedding_dim
 
-
 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: Optional[int] = 4,
         combiner: str = "mean",
         padding_idx: Optional[int] = 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,
@@ -47,23 +49,20 @@ 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 = 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,
-    ):
-
+    def __init__(self, 
+                 name: str, 
+                 vocab_size: int, 
+                 embedding_name: str = '', 
+                 embedding_dim: int = 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):
+        
         self.name = name
         self.vocab_size = vocab_size
         self.embedding_name = embedding_name or name
@@ -76,9 +75,13 @@ class SparseFeature(BaseFeature):
         self.l2_reg = l2_reg
         self.trainable = trainable
 
-
 class DenseFeature(BaseFeature):
-    def __init__(self, name: str, embedding_dim: int = 1):
+    def __init__(self, 
+                 name: str, 
+                 embedding_dim: int = 1):
 
         self.name = name
         self.embedding_dim = embedding_dim
+
+
+