nextrec 0.1.1__tar.gz → 0.1.3__tar.gz

{nextrec-0.1.1 → nextrec-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextrec
-Version: 0.1.1
+Version: 0.1.3
 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
@@ -20,8 +20,8 @@ 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.24.0
-Requires-Dist: pandas>=2.0.0
+Requires-Dist: numpy>=1.23.0
+Requires-Dist: pandas<2.3.0,>=2.1.0
 Requires-Dist: pyarrow>=12.0.0
 Requires-Dist: scikit-learn>=1.3.0
 Requires-Dist: scipy>=1.10.0
@@ -72,47 +72,13 @@ NextRec is a modern recommendation system framework built on PyTorch, providing
 
 ## Installation
 
-NextRec supports installation via **UV** or traditional **pip/source installation**.
-
-### Option 1: Using UV (Recommended)
-
-UV is a modern, high-performance Python package manager offering fast dependency resolution and installation.
-
-```bash
-git clone https://github.com/zerolovesea/NextRec.git
-cd NextRec
-
-# Install UV if not already installed
-pip install uv
-
-# Create virtual environment and install dependencies
-uv sync
-
-# Activate the virtual environment
-source .venv/bin/activate  # macOS/Linux
-# or
-.venv\Scripts\activate     # Windows
-
-# Install the package in editable mode
-uv pip install -e .
-```
-
-**Note**: Make sure to deactivate any other conda/virtual environments before running `uv sync` to avoid environment conflicts.
-
-### Option 2: Using pip/source installation
-
 ```bash
-git clone https://github.com/zerolovesea/NextRec.git
-cd NextRec
-
-# Install dependencies
-pip install -r requirements.txt
-pip install -r test_requirements.txt
+# release version
+pip install nextrec
 
-# Install the package in editable mode
-pip install -e .
+# pre-release version
+pip install -i https://test.pypi.org/simple/ nextrec
 ```
-
 ---
 
 ## 5-Minute Quick Start

{nextrec-0.1.1 → nextrec-0.1.3}/README.md

@@ -30,47 +30,13 @@ NextRec is a modern recommendation system framework built on PyTorch, providing
 
 ## Installation
 
-NextRec supports installation via **UV** or traditional **pip/source installation**.
-
-### Option 1: Using UV (Recommended)
-
-UV is a modern, high-performance Python package manager offering fast dependency resolution and installation.
-
-```bash
-git clone https://github.com/zerolovesea/NextRec.git
-cd NextRec
-
-# Install UV if not already installed
-pip install uv
-
-# Create virtual environment and install dependencies
-uv sync
-
-# Activate the virtual environment
-source .venv/bin/activate  # macOS/Linux
-# or
-.venv\Scripts\activate     # Windows
-
-# Install the package in editable mode
-uv pip install -e .
-```
-
-**Note**: Make sure to deactivate any other conda/virtual environments before running `uv sync` to avoid environment conflicts.
-
-### Option 2: Using pip/source installation
-
 ```bash
-git clone https://github.com/zerolovesea/NextRec.git
-cd NextRec
-
-# Install dependencies
-pip install -r requirements.txt
-pip install -r test_requirements.txt
+# release version
+pip install nextrec
 
-# Install the package in editable mode
-pip install -e .
+# pre-release version
+pip install -i https://test.pypi.org/simple/ nextrec
 ```
-
 ---
 
 ## 5-Minute Quick Start

{nextrec-0.1.1 → nextrec-0.1.3}/README_zh.md

@@ -31,47 +31,12 @@ NextRec 是一个基于 PyTorch 构建的现代推荐系统框架，为研究人
 
 ## 安装
 
-NextRec 提供两种主流的安装方式UV与传统 pip/source 安装。
-
-#### 方法一:使用 UV(推荐)
-
-UV 是一款高速、现代化的 Python 包管理器,能够带来更快的依赖解析与安装体验。
-
 ```bash
-git clone https://github.com/zerolovesea/NextRec.git
-cd NextRec
-
-# 如果尚未安装 UV,先安装它
-pip install uv
-
-# 创建虚拟环境并安装依赖
-uv sync
-
-# 激活虚拟环境
-source .venv/bin/activate  # macOS/Linux
-# 或
-.venv\Scripts\activate     # Windows
-
-# 以可编辑模式安装包
-uv pip install -e .
-```
-
-**注意**: 在运行 `uv sync` 之前,请确保退出其他 conda/虚拟环境,以避免环境冲突。
-
-
-
-#### 方法二:使用 pip/source 
-
-```bash
-git clone https://github.com/zerolovesea/NextRec.git
-cd NextRec
-
-# 安装依赖
-pip install -r requirements.txt
-pip install -r test_requirements.txt
+# 正式版
+pip install nextrec
 
-# 以可编辑模式安装包
-pip install -e .
+# 测试版
+pip install -i https://test.pypi.org/simple/ nextrec
 ```
 
 ## 5分钟快速上手

{nextrec-0.1.1 → nextrec-0.1.3}/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'))
 
-project = 'NextRec'
-copyright = '2025, Yang Zhou'
-author = 'Yang Zhou'
-release = '0.1.1'
+sys.path.insert(0, os.path.abspath("../nextrec"))
+
+project = "NextRec"
+copyright = "2025, Yang Zhou"
+author = "Yang Zhou"
+release = "0.1.3"
 
 # -- 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.3/format_codes.sh

@@ -0,0 +1,9 @@
+#!/bin/zsh
+
+pip install -q black flake8
+
+echo "Using black to format and lint Python files."
+black .
+
+echo "Using flake8 to lint all Python files..."
+flake8 .

{nextrec-0.1.1 → nextrec-0.1.3}/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.3/nextrec/__version__.py

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

{nextrec-0.1.1 → nextrec-0.1.3}/nextrec/basic/activation.py

@@ -14,40 +14,41 @@ 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")
@@ -89,4 +90,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.1 → nextrec-0.1.3}/nextrec/basic/callback.py

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