nextrec 0.3.5__tar.gz → 0.3.6__tar.gz

{nextrec-0.3.5 → nextrec-0.3.6}/.gitignore

@@ -116,6 +116,7 @@ dmypy.json
 
 # test files
 *iflytek*
+generate_feature_config.py
 
 # DataProcessor saved files
 processor/

{nextrec-0.3.5 → nextrec-0.3.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextrec
-Version: 0.3.5
+Version: 0.3.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
@@ -63,7 +63,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.3.5-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.3.6-orange.svg)
 
 English | [中文文档](README_zh.md)
 
@@ -110,7 +110,7 @@ To dive deeper, Jupyter notebooks are available:
 - [Hands on the NextRec framework](/tutorials/notebooks/en/Hands%20on%20nextrec.ipynb)
 - [Using the data processor for preprocessing](/tutorials/notebooks/en/Hands%20on%20dataprocessor.ipynb)
 
-> Current version [0.3.5]: the matching module is not fully polished yet and may have compatibility issues or unexpected errors. Please raise an issue if you run into problems.
+> Current version [0.3.6]: the matching module is not fully polished yet and may have compatibility issues or unexpected errors. Please raise an issue if you run into problems.
 
 ## 5-Minute Quick Start
 

{nextrec-0.3.5 → nextrec-0.3.6}/README.md

@@ -7,7 +7,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.3.5-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.3.6-orange.svg)
 
 English | [中文文档](README_zh.md)
 
@@ -54,7 +54,7 @@ To dive deeper, Jupyter notebooks are available:
 - [Hands on the NextRec framework](/tutorials/notebooks/en/Hands%20on%20nextrec.ipynb)
 - [Using the data processor for preprocessing](/tutorials/notebooks/en/Hands%20on%20dataprocessor.ipynb)
 
-> Current version [0.3.5]: the matching module is not fully polished yet and may have compatibility issues or unexpected errors. Please raise an issue if you run into problems.
+> Current version [0.3.6]: the matching module is not fully polished yet and may have compatibility issues or unexpected errors. Please raise an issue if you run into problems.
 
 ## 5-Minute Quick Start
 

{nextrec-0.3.5 → nextrec-0.3.6}/README_zh.md

@@ -7,7 +7,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.3.5-orange.svg)
+![Version](https://img.shields.io/badge/Version-0.3.6-orange.svg)
 
 [English Version](README.md) | 中文文档
 
@@ -54,7 +54,7 @@ NextRec采用模块化、低耦合的工程设计，使得推荐系统从数据
 - [如何上手NextRec框架](/tutorials/notebooks/zh/Hands%20on%20nextrec.ipynb)
 - [如何使用数据处理器进行数据预处理](/tutorials/notebooks/zh/Hands%20on%20dataprocessor.ipynb)
 
-> 当前版本[0.3.5]，召回模型模块尚不完善，可能存在一些兼容性问题或意外报错，如果遇到问题，欢迎开发者在Issue区提出问题。
+> 当前版本[0.3.6]，召回模型模块尚不完善，可能存在一些兼容性问题或意外报错，如果遇到问题，欢迎开发者在Issue区提出问题。
 
 ## 5分钟快速上手
 

{nextrec-0.3.5 → nextrec-0.3.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.3.5"
+release = "0.3.6"
 
 extensions = [
     "myst_parser",

nextrec-0.3.6/nextrec/__init__.py

@@ -0,0 +1,11 @@
+from nextrec.__version__ import __version__
+
+__all__ = [
+    '__version__',
+]
+
+# Package metadata
+__author__ = "zerolovesea"
+__email__ = "zyaztec@gmail.com"
+__license__ = "Apache 2.0"
+__url__ = "https://github.com/zerolovesea/NextRec"

nextrec-0.3.6/nextrec/__version__.py

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

{nextrec-0.3.5 → nextrec-0.3.6}/nextrec/basic/loggers.py

@@ -99,7 +99,7 @@ def setup_logger(session_id: str | os.PathLike | None = None):
     session = create_session(str(session_id) if session_id is not None else None)
     log_dir = session.logs_dir
     log_dir.mkdir(parents=True, exist_ok=True)
-    log_file = log_dir / f"{session.experiment_id}.log"
+    log_file = log_dir / f"{session.log_basename}.log"
 
     console_format = '%(message)s'
     file_format = '%(asctime)s - %(levelname)s - %(message)s'

{nextrec-0.3.5 → nextrec-0.3.6}/nextrec/basic/model.py

@@ -155,7 +155,7 @@ class BaseModel(FeatureSet, nn.Module):
                 raise ValueError("[BaseModel-input Error] Labels are required but none were found in the input batch.")
         return X_input, y
 
-    def handle_validation_split(self, train_data: dict | pd.DataFrame, validation_split: float, batch_size: int, shuffle: bool,) -> tuple[DataLoader, dict | pd.DataFrame]:
+    def handle_validation_split(self, train_data: dict | pd.DataFrame, validation_split: float, batch_size: int, shuffle: bool, num_workers: int = 0,) -> tuple[DataLoader, dict | pd.DataFrame]:
         """This function will split training data into training and validation sets when: 1. valid_data is None; 2. validation_split is provided."""
         if not (0 < validation_split < 1):
             raise ValueError(f"[BaseModel-validation Error] validation_split must be between 0 and 1, got {validation_split}")
@@ -184,7 +184,7 @@ class BaseModel(FeatureSet, nn.Module):
                 arr = np.asarray(value)      
                 train_split[key] = arr[train_indices]
                 valid_split[key] = arr[valid_indices]
-        train_loader = self.prepare_data_loader(train_split, batch_size=batch_size, shuffle=shuffle)
+        train_loader = self.prepare_data_loader(train_split, batch_size=batch_size, shuffle=shuffle, num_workers=num_workers)
         logging.info(f"Split data: {len(train_indices)} training samples, {len(valid_indices)} validation samples")
         return train_loader, valid_split
 
@@ -265,14 +265,14 @@ class BaseModel(FeatureSet, nn.Module):
                 task_losses.append(task_loss)
             return torch.stack(task_losses).sum()
 
-    def prepare_data_loader(self, data: dict | pd.DataFrame | DataLoader, batch_size: int = 32, shuffle: bool = True,):
+    def prepare_data_loader(self, data: dict | pd.DataFrame | DataLoader, batch_size: int = 32, shuffle: bool = True, num_workers: int = 0,) -> DataLoader:
         if isinstance(data, DataLoader):
             return data
         tensors = build_tensors_from_data(data=data, raw_data=data, features=self.all_features, target_columns=self.target_columns, id_columns=self.id_columns,)
         if tensors is None:
             raise ValueError("[BaseModel-prepare_data_loader Error] No data available to create DataLoader.")
         dataset = TensorDictDataset(tensors)
-        return DataLoader(dataset, batch_size=batch_size, shuffle=shuffle, collate_fn=collate_fn)
+        return DataLoader(dataset, batch_size=batch_size, shuffle=shuffle, collate_fn=collate_fn, num_workers=num_workers)
 
     def fit(self, 
             train_data: dict | pd.DataFrame | DataLoader, 
@@ -281,6 +281,7 @@ class BaseModel(FeatureSet, nn.Module):
             epochs:int=1, shuffle:bool=True, batch_size:int=32,
             user_id_column: str | None = None,
             validation_split: float | None = None,
+            num_workers: int = 0,
             tensorboard: bool = True,):
         self.to(self.device)
         if not self.logger_initialized:
@@ -297,11 +298,11 @@ class BaseModel(FeatureSet, nn.Module):
         self.best_metric = float('-inf') if self.best_metrics_mode == 'max' else float('inf')
 
         if validation_split is not None and valid_data is None:
-            train_loader, valid_data = self.handle_validation_split(train_data=train_data, validation_split=validation_split, batch_size=batch_size, shuffle=shuffle,) # type: ignore
+            train_loader, valid_data = self.handle_validation_split(train_data=train_data, validation_split=validation_split, batch_size=batch_size, shuffle=shuffle, num_workers=num_workers) # type: ignore
         else:
-            train_loader = (train_data if isinstance(train_data, DataLoader) else self.prepare_data_loader(train_data, batch_size=batch_size, shuffle=shuffle))
+            train_loader = (train_data if isinstance(train_data, DataLoader) else self.prepare_data_loader(train_data, batch_size=batch_size, shuffle=shuffle, num_workers=num_workers))
         
-        valid_loader, valid_user_ids = self.prepare_validation_data(valid_data=valid_data, batch_size=batch_size, needs_user_ids=self.needs_user_ids, user_id_column=user_id_column)
+        valid_loader, valid_user_ids = self.prepare_validation_data(valid_data=valid_data, batch_size=batch_size, needs_user_ids=self.needs_user_ids, user_id_column=user_id_column, num_workers=num_workers)
         try:
             self.steps_per_epoch = len(train_loader)
             is_streaming = False
@@ -388,7 +389,7 @@ class BaseModel(FeatureSet, nn.Module):
                 self.training_logger.log_metrics(train_log_payload, step=epoch + 1, split="train")
             if valid_loader is not None:
                 # pass user_ids only if needed for GAUC metric
-                val_metrics = self.evaluate(valid_loader, user_ids=valid_user_ids if self.needs_user_ids else None) # {'auc': 0.75, 'logloss': 0.45} or {'auc_target1': 0.75, 'logloss_target1': 0.45, 'mse_target2': 3.2}
+                val_metrics = self.evaluate(valid_loader, user_ids=valid_user_ids if self.needs_user_ids else None, num_workers=num_workers) # {'auc': 0.75, 'logloss': 0.45} or {'auc_target1': 0.75, 'logloss_target1': 0.45, 'mse_target2': 3.2}
                 if self.nums_task == 1:
                     metrics_str = ", ".join([f"{k}={v:.4f}" for k, v in val_metrics.items()])
                     logging.info(colorize(f"  Epoch {epoch + 1}/{epochs} - Valid: {metrics_str}", color="cyan"))
@@ -513,12 +514,12 @@ class BaseModel(FeatureSet, nn.Module):
             return avg_loss, metrics_dict
         return avg_loss
 
-    def prepare_validation_data(self, valid_data: dict | pd.DataFrame | DataLoader | None, batch_size: int, needs_user_ids: bool, user_id_column: str | None = 'user_id') -> tuple[DataLoader | None, np.ndarray | None]:
+    def prepare_validation_data(self, valid_data: dict | pd.DataFrame | DataLoader | None, batch_size: int, needs_user_ids: bool, user_id_column: str | None = 'user_id', num_workers: int = 0,) -> tuple[DataLoader | None, np.ndarray | None]:
         if valid_data is None:
             return None, None
         if isinstance(valid_data, DataLoader):
             return valid_data, None
-        valid_loader = self.prepare_data_loader(valid_data, batch_size=batch_size, shuffle=False)
+        valid_loader = self.prepare_data_loader(valid_data, batch_size=batch_size, shuffle=False, num_workers=num_workers)
         valid_user_ids = None
         if needs_user_ids:
             if user_id_column is None:
@@ -531,7 +532,8 @@ class BaseModel(FeatureSet, nn.Module):
                  metrics: list[str] | dict[str, list[str]] | None = None,
                  batch_size: int = 32,
                  user_ids: np.ndarray | None = None,
-                 user_id_column: str = 'user_id') -> dict:
+                 user_id_column: str = 'user_id',
+                 num_workers: int = 0,) -> dict:
         self.eval()
         eval_metrics = metrics if metrics is not None else self.metrics
         if eval_metrics is None:
@@ -543,7 +545,7 @@ class BaseModel(FeatureSet, nn.Module):
         else:
             if user_ids is None and needs_user_ids:
                 user_ids = get_user_ids(data=data, id_columns=user_id_column)
-            data_loader = self.prepare_data_loader(data, batch_size=batch_size, shuffle=False)
+            data_loader = self.prepare_data_loader(data, batch_size=batch_size, shuffle=False, num_workers=num_workers)
         y_true_list = []
         y_pred_list = []
         collected_user_ids = []
@@ -603,6 +605,7 @@ class BaseModel(FeatureSet, nn.Module):
         include_ids: bool | None = None,
         return_dataframe: bool = True,
         streaming_chunk_size: int = 10000,
+        num_workers: int = 0,
     ) -> pd.DataFrame | np.ndarray:
         self.eval()
         if include_ids is None:
@@ -615,7 +618,7 @@ class BaseModel(FeatureSet, nn.Module):
             rec_loader = RecDataLoader(dense_features=self.dense_features, sparse_features=self.sparse_features, sequence_features=self.sequence_features, target=self.target_columns, id_columns=self.id_columns,)
             data_loader = rec_loader.create_dataloader(data=data, batch_size=batch_size, shuffle=False, load_full=False, chunk_size=streaming_chunk_size,)
         elif not isinstance(data, DataLoader):
-            data_loader = self.prepare_data_loader(data, batch_size=batch_size, shuffle=False,)
+            data_loader = self.prepare_data_loader(data, batch_size=batch_size, shuffle=False, num_workers=num_workers)
         else:
             data_loader = data
         

{nextrec-0.3.5 → nextrec-0.3.6}/nextrec/basic/session.py

@@ -22,6 +22,7 @@ class Session:
 
     experiment_id: str
     root: Path
+    log_basename: str  # The base name for log files, without path separators
 
     @property
     def logs_dir(self) -> Path:
@@ -60,7 +61,6 @@ class Session:
         return path
 
 def create_session(experiment_id: str | Path | None = None) -> Session:
-    """Create a :class:`Session` instance with prepared directories."""
 
     if experiment_id is not None and str(experiment_id).strip():
         exp_id = str(experiment_id).strip()
@@ -68,6 +68,8 @@ def create_session(experiment_id: str | Path | None = None) -> Session:
         # Use local time for session naming
         exp_id = "nextrec_session_" + datetime.now().strftime("%Y%m%d")
 
+    log_basename = Path(exp_id).name if exp_id else exp_id
+
     if (
         os.getenv("PYTEST_CURRENT_TEST")
         or os.getenv("PYTEST_RUNNING")
@@ -82,7 +84,7 @@ def create_session(experiment_id: str | Path | None = None) -> Session:
     session_path.mkdir(parents=True, exist_ok=True)
     root = session_path.resolve()
 
-    return Session(experiment_id=exp_id, root=root)
+    return Session(experiment_id=exp_id, root=root, log_basename=log_basename)
 
 def resolve_save_path(
     path: str | os.PathLike | Path | None,

{nextrec-0.3.5 → nextrec-0.3.6}/nextrec/data/__init__.py

@@ -1,22 +1,4 @@
-"""
-Data utilities package for NextRec
-
-This package provides data processing and manipulation utilities organized by category:
-- batch_utils: Batch collation and processing
-- data_processing: Data manipulation and user ID extraction
-- data_utils: Legacy module (re-exports from specialized modules)
-- dataloader: Dataset and DataLoader implementations
-- preprocessor: Data preprocessing pipeline
-
-Date: create on 13/11/2025
-Last update: 03/12/2025 (refactored)
-Author: Yang Zhou, zyaztec@gmail.com
-"""
-
-# Batch utilities
 from nextrec.data.batch_utils import collate_fn, batch_to_dict, stack_section
-
-# Data processing utilities
 from nextrec.data.data_processing import (
     get_column_data,
     split_dict_random,
@@ -24,7 +6,6 @@ from nextrec.data.data_processing import (
     get_user_ids,
 )
 
-# File utilities (from utils package)
 from nextrec.utils.file import (
     resolve_file_paths,
     iter_file_chunks,
@@ -33,7 +14,6 @@ from nextrec.utils.file import (
     default_output_dir,
 )
 
-# DataLoader components
 from nextrec.data.dataloader import (
     TensorDictDataset,
     FileDataset,
@@ -41,13 +21,8 @@ from nextrec.data.dataloader import (
     build_tensors_from_data,
 )
 
-# Preprocessor
 from nextrec.data.preprocessor import DataProcessor
-
-# Feature definitions
 from nextrec.basic.features import FeatureSet
-
-# Legacy module (for backward compatibility)
 from nextrec.data import data_utils
 
 __all__ = [

{nextrec-0.3.5 → nextrec-0.3.6}/nextrec/data/dataloader.py

@@ -126,20 +126,22 @@ class RecDataLoader(FeatureSet):
                          batch_size: int = 32,
                          shuffle: bool = True,
                          load_full: bool = True,
-                         chunk_size: int = 10000) -> DataLoader:
+                         chunk_size: int = 10000,
+                         num_workers: int = 0) -> DataLoader:
         if isinstance(data, DataLoader):
             return data
         elif isinstance(data, (str, os.PathLike)):
-            return self.create_from_path(path=data, batch_size=batch_size, shuffle=shuffle, load_full=load_full, chunk_size=chunk_size)
+            return self.create_from_path(path=data, batch_size=batch_size, shuffle=shuffle, load_full=load_full, chunk_size=chunk_size, num_workers=num_workers)
         elif isinstance(data, (dict, pd.DataFrame)):
-            return self.create_from_memory(data=data, batch_size=batch_size, shuffle=shuffle)
+            return self.create_from_memory(data=data, batch_size=batch_size, shuffle=shuffle, num_workers=num_workers)
         else:
             raise ValueError(f"[RecDataLoader Error] Unsupported data type: {type(data)}")
     
     def create_from_memory(self, 
                            data: dict | pd.DataFrame,
                            batch_size: int,
-                           shuffle: bool) -> DataLoader:
+                           shuffle: bool,
+                           num_workers: int = 0) -> DataLoader:
         raw_data = data
 
         if self.processor is not None:
@@ -150,14 +152,15 @@ class RecDataLoader(FeatureSet):
         if tensors is None:
             raise ValueError("[RecDataLoader Error] No valid tensors could be built from the provided data.")
         dataset = TensorDictDataset(tensors)
-        return DataLoader(dataset, batch_size=batch_size, shuffle=shuffle, collate_fn=collate_fn)
+        return DataLoader(dataset, batch_size=batch_size, shuffle=shuffle, collate_fn=collate_fn, num_workers=num_workers)
     
     def create_from_path(self,
                          path: str,
                          batch_size: int,
                          shuffle: bool,
                          load_full: bool,
-                         chunk_size: int = 10000) -> DataLoader:
+                         chunk_size: int = 10000,
+                         num_workers: int = 0) -> DataLoader:
         file_paths, file_type = resolve_file_paths(str(Path(path)))
         # Load full data into memory
         if load_full:
@@ -169,6 +172,7 @@ class RecDataLoader(FeatureSet):
                 except OSError:
                     pass
                 try:
+                    df = read_table(file_path, file_type=file_type)
                     dfs.append(df)
                 except MemoryError as exc:
                     raise MemoryError(f"[RecDataLoader Error] Out of memory while reading {file_path}. Consider using load_full=False with streaming.") from exc
@@ -176,22 +180,23 @@ class RecDataLoader(FeatureSet):
                 combined_df = pd.concat(dfs, ignore_index=True)
             except MemoryError as exc:
                 raise MemoryError(f"[RecDataLoader Error] Out of memory while concatenating loaded data (approx {total_bytes / (1024**3):.2f} GB). Use load_full=False to stream or reduce chunk_size.") from exc
-            return self.create_from_memory(combined_df, batch_size, shuffle,)
+            return self.create_from_memory(combined_df, batch_size, shuffle, num_workers=num_workers)
         else:
-            return self.load_files_streaming(file_paths, file_type, batch_size, chunk_size, shuffle)
+            return self.load_files_streaming(file_paths, file_type, batch_size, chunk_size, shuffle, num_workers=num_workers)
 
     def load_files_streaming(self,
                              file_paths: list[str],
                              file_type: str,
                              batch_size: int,
                              chunk_size: int,
-                             shuffle: bool) -> DataLoader:
+                             shuffle: bool,
+                             num_workers: int = 0) -> DataLoader:
         if shuffle:
             logging.info("[RecDataLoader Info] Shuffle is ignored in streaming mode (IterableDataset).")
         if batch_size != 1:
             logging.info("[RecDataLoader Info] Streaming mode enforces batch_size=1; tune chunk_size to control memory/throughput.")
         dataset = FileDataset(file_paths=file_paths, dense_features=self.dense_features, sparse_features=self.sparse_features, sequence_features=self.sequence_features, target_columns=self.target_columns, id_columns=self.id_columns, chunk_size=chunk_size, file_type=file_type, processor=self.processor)
-        return DataLoader(dataset, batch_size=1, collate_fn=collate_fn)
+        return DataLoader(dataset, batch_size=1, collate_fn=collate_fn, num_workers=num_workers)
 
 def normalize_sequence_column(column, feature: SequenceFeature) -> np.ndarray:
     if isinstance(column, pd.Series):

{nextrec-0.3.5 → nextrec-0.3.6}/nextrec/utils/device.py

@@ -4,9 +4,10 @@ Device management utilities for NextRec
 Date: create on 03/12/2025
 Author: Yang Zhou, zyaztec@gmail.com
 """
-
+import os
 import torch
 import platform
+import multiprocessing
 
 
 def resolve_device() -> str:

{nextrec-0.3.5 → nextrec-0.3.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "nextrec"
-version = "0.3.5"
+version = "0.3.6"
 description = "A comprehensive recommendation library with match, ranking, and multi-task learning models"
 readme = "README.md"
 requires-python = ">=3.10"

nextrec-0.3.5/nextrec/__init__.py

@@ -1,41 +0,0 @@
-"""
-NextRec - A Unified Deep Learning Framework for Recommender Systems
-===================================================================
-
-NextRec provides a comprehensive suite of recommendation models including:
-- Ranking models (CTR prediction)
-- Matching models (retrieval)
-- Multi-task learning models
-- Generative recommendation models
-
-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)
-"""
-
-from nextrec.__version__ import __version__
-
-__all__ = [
-    '__version__',
-]
-
-# Package metadata
-__author__ = "zerolovesea"
-__email__ = "zyaztec@gmail.com"
-__license__ = "Apache 2.0"
-__url__ = "https://github.com/zerolovesea/NextRec"

nextrec-0.3.5/nextrec/__version__.py

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

nextrec-0.3.5/nextrec/models/generative/__init__.py

@@ -1,5 +0,0 @@
-from .hstu import HSTU
-
-__all__ = [
-    "HSTU",
-]

nextrec-0.3.5/nextrec/models/match/__init__.py

@@ -1,13 +0,0 @@
-from .dssm import DSSM
-from .dssm_v2 import DSSM_v2
-from .youtube_dnn import YoutubeDNN
-from .mind import MIND
-from .sdm import SDM
-
-__all__ = [
-    'DSSM',
-    'DSSM_v2',
-    'YoutubeDNN',
-    'MIND',
-    'SDM',
-]

nextrec-0.3.5/nextrec/models/ranking/__init__.py

@@ -1,27 +0,0 @@
-from .fm import FM
-from .afm import AFM
-from .masknet import MaskNet
-from .pnn import PNN
-from .deepfm import DeepFM
-from .autoint import AutoInt
-from .widedeep import WideDeep
-from .xdeepfm import xDeepFM
-from .dcn import DCN
-from .fibinet import FiBiNET
-from .din import DIN
-from .dien import DIEN
-
-__all__ = [
-    'DeepFM',
-    'AutoInt', 
-    'WideDeep',
-    'xDeepFM',
-    'DCN',
-    'DIN',
-    'DIEN',
-    'FM',
-    'AFM',
-    'MaskNet',
-    'PNN',
-    'FiBiNET',
-]