nextrec 0.1.4__tar.gz → 0.1.7__tar.gz

{nextrec-0.1.4 → nextrec-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextrec
-Version: 0.1.4
+Version: 0.1.7
 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
@@ -19,12 +19,21 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.10
-Requires-Dist: fastparquet>=2023.4.0
-Requires-Dist: numpy>=1.23.0
-Requires-Dist: pandas<2.3.0,>=2.1.0
-Requires-Dist: pyarrow<15.0.0,>=12.0.0
-Requires-Dist: scikit-learn>=1.3.0
-Requires-Dist: scipy>=1.10.0
+Requires-Dist: numpy<2.0,>=1.21; sys_platform == 'linux'
+Requires-Dist: numpy>=1.23.0; sys_platform == 'win32'
+Requires-Dist: numpy>=1.24.0; sys_platform == 'darwin'
+Requires-Dist: pandas<2.0,>=1.5; sys_platform == 'linux'
+Requires-Dist: pandas<2.3.0,>=2.1.0; sys_platform == 'win32'
+Requires-Dist: pandas>=2.0.0; sys_platform == 'darwin'
+Requires-Dist: pyarrow<13.0.0,>=10.0.0; sys_platform == 'linux'
+Requires-Dist: pyarrow<15.0.0,>=12.0.0; sys_platform == 'win32'
+Requires-Dist: pyarrow>=12.0.0; sys_platform == 'darwin'
+Requires-Dist: scikit-learn<2.0,>=1.2; sys_platform == 'linux'
+Requires-Dist: scikit-learn>=1.3.0; sys_platform == 'darwin'
+Requires-Dist: scikit-learn>=1.3.0; sys_platform == 'win32'
+Requires-Dist: scipy<1.12,>=1.8; sys_platform == 'linux'
+Requires-Dist: scipy>=1.10.0; sys_platform == 'darwin'
+Requires-Dist: scipy>=1.10.0; sys_platform == 'win32'
 Requires-Dist: torch>=2.0.0
 Requires-Dist: torchvision>=0.15.0
 Requires-Dist: tqdm>=4.65.0

{nextrec-0.1.4 → nextrec-0.1.7}/docs/conf.py

@@ -7,29 +7,29 @@
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#project-information
 import os
 import sys
-
-sys.path.insert(0, os.path.abspath("../nextrec"))
+sys.path.insert(0, os.path.abspath('../nextrec'))
 
 project = "NextRec"
 copyright = "2025, Yang Zhou"
 author = "Yang Zhou"
-release = "0.1.4"
+release = "0.1.7"
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration
 
 extensions = [
-    "sphinx.ext.autodoc",
-    "sphinx.ext.napoleon",
-    "sphinx_rtd_theme",
+    'sphinx.ext.autodoc',
+    'sphinx.ext.napoleon',
+    'sphinx_rtd_theme',
 ]
 
-templates_path = ["_templates"]
-exclude_patterns = ["_build", "Thumbs.db", ".DS_Store"]
+templates_path = ['_templates']
+exclude_patterns = ['_build', 'Thumbs.db', '.DS_Store']
+
 
 
 # -- Options for HTML output -------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#options-for-html-output
 
-html_theme = "sphinx_rtd_theme"
-html_static_path = ["_static"]
+html_theme = 'sphinx_rtd_theme'
+html_static_path = ['_static']

{nextrec-0.1.4 → nextrec-0.1.7}/nextrec/__init__.py

@@ -12,18 +12,18 @@ Quick Start
 -----------
 >>> from nextrec.basic.features import DenseFeature, SparseFeature
 >>> from nextrec.models.ranking.deepfm import DeepFM
->>>
+>>> 
 >>> # Define features
 >>> dense_features = [DenseFeature('age')]
 >>> sparse_features = [SparseFeature('category', vocab_size=100, embedding_dim=16)]
->>>
+>>> 
 >>> # Build model
 >>> model = DeepFM(
 ...     dense_features=dense_features,
 ...     sparse_features=sparse_features,
 ...     targets=['label']
 ... )
->>>
+>>> 
 >>> # Train model
 >>> model.fit(train_data=df_train, valid_data=df_valid)
 """
@@ -31,7 +31,7 @@ Quick Start
 from nextrec.__version__ import __version__
 
 __all__ = [
-    "__version__",
+    '__version__',
 ]
 
 # Package metadata

nextrec-0.1.7/nextrec/__version__.py

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

{nextrec-0.1.4 → nextrec-0.1.7}/nextrec/basic/activation.py

@@ -14,41 +14,40 @@ 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
             x = x.view(-1, emb_size)
-
+        
         x_norm = self.bn(x)
         p = torch.sigmoid(x_norm)
         output = p * x + (1 - p) * self.alpha * x
-
+        
         if len(original_shape) == 3:
             output = output.view(original_shape)
-
+        
         return output
 
 
 def activation_layer(activation: str, emb_size: int | None = None):
     """Create an activation layer based on the given activation name."""
-
+    
     activation = activation.lower()
-
+    
     if activation == "dice":
         if emb_size is None:
             raise ValueError("emb_size is required for Dice activation")
@@ -90,4 +89,4 @@ def activation_layer(activation: str, emb_size: int | None = None):
     elif activation in ["none", "linear", "identity"]:
         return nn.Identity()
     else:
-        raise ValueError(f"Unsupported activation function: {activation}")
+        raise ValueError(f"Unsupported activation function: {activation}")  

{nextrec-0.1.4 → nextrec-0.1.7}/nextrec/basic/callback.py

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