nextrec 0.4.8__py3-none-any.whl → 0.4.10__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (66) hide show
  1. nextrec/__version__.py +1 -1
  2. nextrec/basic/callback.py +30 -15
  3. nextrec/basic/features.py +1 -0
  4. nextrec/basic/layers.py +6 -8
  5. nextrec/basic/loggers.py +14 -7
  6. nextrec/basic/metrics.py +6 -76
  7. nextrec/basic/model.py +316 -321
  8. nextrec/cli.py +185 -43
  9. nextrec/data/__init__.py +13 -16
  10. nextrec/data/batch_utils.py +3 -2
  11. nextrec/data/data_processing.py +10 -2
  12. nextrec/data/data_utils.py +9 -14
  13. nextrec/data/dataloader.py +31 -33
  14. nextrec/data/preprocessor.py +328 -255
  15. nextrec/loss/__init__.py +1 -5
  16. nextrec/loss/loss_utils.py +2 -8
  17. nextrec/models/generative/__init__.py +1 -8
  18. nextrec/models/generative/hstu.py +6 -4
  19. nextrec/models/multi_task/esmm.py +2 -2
  20. nextrec/models/multi_task/mmoe.py +2 -2
  21. nextrec/models/multi_task/ple.py +2 -2
  22. nextrec/models/multi_task/poso.py +2 -3
  23. nextrec/models/multi_task/share_bottom.py +2 -2
  24. nextrec/models/ranking/afm.py +2 -2
  25. nextrec/models/ranking/autoint.py +2 -2
  26. nextrec/models/ranking/dcn.py +2 -2
  27. nextrec/models/ranking/dcn_v2.py +2 -2
  28. nextrec/models/ranking/deepfm.py +6 -7
  29. nextrec/models/ranking/dien.py +3 -3
  30. nextrec/models/ranking/din.py +3 -3
  31. nextrec/models/ranking/eulernet.py +365 -0
  32. nextrec/models/ranking/fibinet.py +5 -5
  33. nextrec/models/ranking/fm.py +3 -7
  34. nextrec/models/ranking/lr.py +120 -0
  35. nextrec/models/ranking/masknet.py +2 -2
  36. nextrec/models/ranking/pnn.py +2 -2
  37. nextrec/models/ranking/widedeep.py +2 -2
  38. nextrec/models/ranking/xdeepfm.py +2 -2
  39. nextrec/models/representation/__init__.py +9 -0
  40. nextrec/models/{generative → representation}/rqvae.py +9 -9
  41. nextrec/models/retrieval/__init__.py +0 -0
  42. nextrec/models/{match → retrieval}/dssm.py +8 -3
  43. nextrec/models/{match → retrieval}/dssm_v2.py +8 -3
  44. nextrec/models/{match → retrieval}/mind.py +4 -3
  45. nextrec/models/{match → retrieval}/sdm.py +4 -3
  46. nextrec/models/{match → retrieval}/youtube_dnn.py +8 -3
  47. nextrec/utils/__init__.py +60 -46
  48. nextrec/utils/config.py +8 -7
  49. nextrec/utils/console.py +371 -0
  50. nextrec/utils/{synthetic_data.py → data.py} +102 -15
  51. nextrec/utils/feature.py +15 -0
  52. nextrec/utils/torch_utils.py +411 -0
  53. {nextrec-0.4.8.dist-info → nextrec-0.4.10.dist-info}/METADATA +6 -7
  54. nextrec-0.4.10.dist-info/RECORD +70 -0
  55. nextrec/utils/cli_utils.py +0 -58
  56. nextrec/utils/device.py +0 -78
  57. nextrec/utils/distributed.py +0 -141
  58. nextrec/utils/file.py +0 -92
  59. nextrec/utils/initializer.py +0 -79
  60. nextrec/utils/optimizer.py +0 -75
  61. nextrec/utils/tensor.py +0 -72
  62. nextrec-0.4.8.dist-info/RECORD +0 -71
  63. /nextrec/models/{match/__init__.py → ranking/ffm.py} +0 -0
  64. {nextrec-0.4.8.dist-info → nextrec-0.4.10.dist-info}/WHEEL +0 -0
  65. {nextrec-0.4.8.dist-info → nextrec-0.4.10.dist-info}/entry_points.txt +0 -0
  66. {nextrec-0.4.8.dist-info → nextrec-0.4.10.dist-info}/licenses/LICENSE +0 -0
@@ -1,141 +0,0 @@
1
- """
2
- Distributed utilities for NextRec.
3
-
4
- Date: create on 04/12/2025
5
- Checkpoint: edit on 05/12/2025
6
- Author: Yang Zhou,zyaztec@gmail.com
7
- """
8
-
9
- import logging
10
- import numpy as np
11
- import torch
12
- import torch.distributed as dist
13
-
14
- from torch.utils.data import DataLoader, IterableDataset
15
- from torch.utils.data.distributed import DistributedSampler
16
- from nextrec.basic.loggers import colorize
17
-
18
-
19
- def init_process_group(
20
- distributed: bool, rank: int, world_size: int, device_id: int | None = None
21
- ) -> None:
22
- """
23
- initialize distributed process group for multi-GPU training.
24
-
25
- Args:
26
- distributed: whether to enable distributed training
27
- rank: global rank of the current process
28
- world_size: total number of processes
29
- """
30
- if (not distributed) or (not dist.is_available()) or dist.is_initialized():
31
- return
32
- backend = "nccl" if device_id is not None else "gloo"
33
- if backend == "nccl":
34
- torch.cuda.set_device(device_id)
35
- dist.init_process_group(
36
- backend=backend, init_method="env://", rank=rank, world_size=world_size
37
- )
38
-
39
-
40
- def gather_numpy(self, array: np.ndarray | None) -> np.ndarray | None:
41
- """
42
- Gather numpy arrays (or None) across ranks. Uses all_gather_object to avoid
43
- shape mismatches and ensures every rank participates even when local data is empty.
44
- """
45
- if not (self.distributed and dist.is_available() and dist.is_initialized()):
46
- return array
47
-
48
- world_size = dist.get_world_size()
49
- gathered: list[np.ndarray | None] = [None for _ in range(world_size)]
50
- dist.all_gather_object(gathered, array)
51
- pieces: list[np.ndarray] = []
52
- for item in gathered:
53
- if item is None:
54
- continue
55
- item_np = np.asarray(item)
56
- if item_np.size > 0:
57
- pieces.append(item_np)
58
- if not pieces:
59
- return None
60
- return np.concatenate(pieces, axis=0)
61
-
62
-
63
- def add_distributed_sampler(
64
- loader: DataLoader,
65
- distributed: bool,
66
- world_size: int,
67
- rank: int,
68
- shuffle: bool,
69
- drop_last: bool,
70
- default_batch_size: int,
71
- is_main_process: bool = False,
72
- ) -> tuple[DataLoader, DistributedSampler | None]:
73
- """
74
- add distributedsampler to a dataloader, this for distributed training
75
- when each device has its own dataloader
76
- """
77
- # early return if not distributed
78
- if not (distributed and dist.is_available() and dist.is_initialized()):
79
- return loader, None
80
- # return if already has DistributedSampler
81
- if isinstance(loader.sampler, DistributedSampler):
82
- return loader, loader.sampler
83
- dataset = getattr(loader, "dataset", None)
84
- if dataset is None:
85
- return loader, None
86
- if isinstance(dataset, IterableDataset):
87
- if is_main_process:
88
- logging.info(
89
- colorize(
90
- "[Distributed Info] Iterable/streaming DataLoader provided; DistributedSampler is skipped. Ensure dataset handles sharding per rank.",
91
- color="yellow",
92
- )
93
- )
94
- return loader, None
95
- sampler = DistributedSampler(
96
- dataset,
97
- num_replicas=world_size,
98
- rank=rank,
99
- shuffle=shuffle,
100
- drop_last=drop_last,
101
- )
102
- loader_kwargs = {
103
- "batch_size": (
104
- loader.batch_size if loader.batch_size is not None else default_batch_size
105
- ),
106
- "shuffle": False,
107
- "sampler": sampler,
108
- "num_workers": loader.num_workers,
109
- "collate_fn": loader.collate_fn,
110
- "drop_last": drop_last,
111
- }
112
- if getattr(loader, "pin_memory", False):
113
- loader_kwargs["pin_memory"] = True
114
- pin_memory_device = getattr(loader, "pin_memory_device", None)
115
- if pin_memory_device:
116
- loader_kwargs["pin_memory_device"] = pin_memory_device
117
- timeout = getattr(loader, "timeout", None)
118
- if timeout:
119
- loader_kwargs["timeout"] = timeout
120
- worker_init_fn = getattr(loader, "worker_init_fn", None)
121
- if worker_init_fn is not None:
122
- loader_kwargs["worker_init_fn"] = worker_init_fn
123
- generator = getattr(loader, "generator", None)
124
- if generator is not None:
125
- loader_kwargs["generator"] = generator
126
- if loader.num_workers > 0:
127
- loader_kwargs["persistent_workers"] = getattr(
128
- loader, "persistent_workers", False
129
- )
130
- prefetch_factor = getattr(loader, "prefetch_factor", None)
131
- if prefetch_factor is not None:
132
- loader_kwargs["prefetch_factor"] = prefetch_factor
133
- distributed_loader = DataLoader(dataset, **loader_kwargs)
134
- if is_main_process:
135
- logging.info(
136
- colorize(
137
- "[Distributed Info] Attached DistributedSampler to provided DataLoader",
138
- color="cyan",
139
- )
140
- )
141
- return distributed_loader, sampler
nextrec/utils/file.py DELETED
@@ -1,92 +0,0 @@
1
- """
2
- File I/O utilities for NextRec
3
-
4
- Date: create on 03/12/2025
5
- Checkpoint: edit on 06/12/2025
6
- Author: Yang Zhou, zyaztec@gmail.com
7
- """
8
-
9
- import yaml
10
- import pandas as pd
11
- import pyarrow.parquet as pq
12
-
13
- from pathlib import Path
14
- from typing import Generator
15
-
16
-
17
- def resolve_file_paths(path: str) -> tuple[list[str], str]:
18
- """
19
- Resolve file or directory path into a sorted list of files and file type.
20
-
21
- Args: path: Path to a file or directory
22
- Returns: tuple: (list of file paths, file type)
23
- """
24
- path_obj = Path(path)
25
-
26
- if path_obj.is_file():
27
- file_type = path_obj.suffix.lower().lstrip(".")
28
- assert file_type in [
29
- "csv",
30
- "parquet",
31
- ], f"Unsupported file extension: {file_type}"
32
- return [str(path_obj)], file_type
33
-
34
- if path_obj.is_dir():
35
- collected_files = [p for p in path_obj.iterdir() if p.is_file()]
36
- csv_files = [str(p) for p in collected_files if p.suffix.lower() == ".csv"]
37
- parquet_files = [
38
- str(p) for p in collected_files if p.suffix.lower() == ".parquet"
39
- ]
40
-
41
- if csv_files and parquet_files:
42
- raise ValueError(
43
- "Directory contains both CSV and Parquet files. Please keep a single format."
44
- )
45
- file_paths = csv_files if csv_files else parquet_files
46
- if not file_paths:
47
- raise ValueError(f"No CSV or Parquet files found in directory: {path}")
48
- file_paths.sort()
49
- file_type = "csv" if csv_files else "parquet"
50
- return file_paths, file_type
51
-
52
- raise ValueError(f"Invalid path: {path}")
53
-
54
-
55
- def read_table(path: str | Path, data_format: str | None = None) -> pd.DataFrame:
56
- data_path = Path(path)
57
- fmt = data_format.lower() if data_format else data_path.suffix.lower().lstrip(".")
58
- if data_path.is_dir() and not fmt:
59
- fmt = "parquet"
60
- if fmt in {"parquet", ""}:
61
- return pd.read_parquet(data_path)
62
- if fmt in {"csv", "txt"}:
63
- # Use low_memory=False to avoid mixed-type DtypeWarning on wide CSVs
64
- return pd.read_csv(data_path, low_memory=False)
65
- raise ValueError(f"Unsupported data format: {data_path}")
66
-
67
-
68
- def load_dataframes(file_paths: list[str], file_type: str) -> list[pd.DataFrame]:
69
- return [read_table(fp, file_type) for fp in file_paths]
70
-
71
-
72
- def iter_file_chunks(
73
- file_path: str, file_type: str, chunk_size: int
74
- ) -> Generator[pd.DataFrame, None, None]:
75
- if file_type == "csv":
76
- yield from pd.read_csv(file_path, chunksize=chunk_size)
77
- return
78
- parquet_file = pq.ParquetFile(file_path)
79
- for batch in parquet_file.iter_batches(batch_size=chunk_size):
80
- yield batch.to_pandas()
81
-
82
-
83
- def default_output_dir(path: str) -> Path:
84
- path_obj = Path(path)
85
- if path_obj.is_file():
86
- return path_obj.parent / f"{path_obj.stem}_preprocessed"
87
- return path_obj.with_name(f"{path_obj.name}_preprocessed")
88
-
89
-
90
- def read_yaml(path: str | Path):
91
- with open(path, "r", encoding="utf-8") as file:
92
- return yaml.safe_load(file) or {}
@@ -1,79 +0,0 @@
1
- """
2
- Initialization utilities for NextRec
3
-
4
- Date: create on 13/11/2025
5
- Author: Yang Zhou, zyaztec@gmail.com
6
- """
7
-
8
- from typing import Any, Dict, Set
9
-
10
- import torch.nn as nn
11
-
12
- KNOWN_NONLINEARITIES: Set[str] = {
13
- "linear",
14
- "conv1d",
15
- "conv2d",
16
- "conv3d",
17
- "conv_transpose1d",
18
- "conv_transpose2d",
19
- "conv_transpose3d",
20
- "sigmoid",
21
- "tanh",
22
- "relu",
23
- "leaky_relu",
24
- "selu",
25
- "gelu",
26
- }
27
-
28
-
29
- def resolve_nonlinearity(activation: str):
30
- if activation in KNOWN_NONLINEARITIES:
31
- return activation
32
- return "linear"
33
-
34
-
35
- def resolve_gain(activation: str, param: Dict[str, Any]) -> float:
36
- if "gain" in param:
37
- return param["gain"]
38
- nonlinearity = resolve_nonlinearity(activation)
39
- try:
40
- return nn.init.calculate_gain(nonlinearity, param.get("param")) # type: ignore
41
- except ValueError:
42
- return 1.0
43
-
44
-
45
- def get_initializer(
46
- init_type: str = "normal",
47
- activation: str = "linear",
48
- param: Dict[str, Any] | None = None,
49
- ):
50
- param = param or {}
51
- nonlinearity = resolve_nonlinearity(activation)
52
- gain = resolve_gain(activation, param)
53
-
54
- def initializer_fn(tensor):
55
- if init_type == "xavier_uniform":
56
- nn.init.xavier_uniform_(tensor, gain=gain)
57
- elif init_type == "xavier_normal":
58
- nn.init.xavier_normal_(tensor, gain=gain)
59
- elif init_type == "kaiming_uniform":
60
- nn.init.kaiming_uniform_(
61
- tensor, a=param.get("a", 0), nonlinearity=nonlinearity # type: ignore
62
- )
63
- elif init_type == "kaiming_normal":
64
- nn.init.kaiming_normal_(
65
- tensor, a=param.get("a", 0), nonlinearity=nonlinearity # type: ignore
66
- )
67
- elif init_type == "orthogonal":
68
- nn.init.orthogonal_(tensor, gain=gain)
69
- elif init_type == "normal":
70
- nn.init.normal_(
71
- tensor, mean=param.get("mean", 0.0), std=param.get("std", 0.0001)
72
- )
73
- elif init_type == "uniform":
74
- nn.init.uniform_(tensor, a=param.get("a", -0.05), b=param.get("b", 0.05))
75
- else:
76
- raise ValueError(f"Unknown init_type: {init_type}")
77
- return tensor
78
-
79
- return initializer_fn
@@ -1,75 +0,0 @@
1
- """
2
- Optimizer and Scheduler utilities for NextRec
3
-
4
- Date: create on 13/11/2025
5
- Author: Yang Zhou, zyaztec@gmail.com
6
- """
7
-
8
- import torch
9
- from typing import Iterable
10
-
11
-
12
- def get_optimizer(
13
- optimizer: str | torch.optim.Optimizer = "adam",
14
- params: Iterable[torch.nn.Parameter] | None = None,
15
- **optimizer_params,
16
- ):
17
- if params is None:
18
- raise ValueError("params cannot be None. Please provide model parameters.")
19
-
20
- if "lr" not in optimizer_params:
21
- optimizer_params["lr"] = 1e-3
22
- if isinstance(optimizer, str):
23
- opt_name = optimizer.lower()
24
- if opt_name == "adam":
25
- opt_class = torch.optim.Adam
26
- elif opt_name == "sgd":
27
- opt_class = torch.optim.SGD
28
- elif opt_name == "adamw":
29
- opt_class = torch.optim.AdamW
30
- elif opt_name == "adagrad":
31
- opt_class = torch.optim.Adagrad
32
- elif opt_name == "rmsprop":
33
- opt_class = torch.optim.RMSprop
34
- else:
35
- raise NotImplementedError(f"Unsupported optimizer: {optimizer}")
36
- optimizer_fn = opt_class(params=params, **optimizer_params)
37
- elif isinstance(optimizer, torch.optim.Optimizer):
38
- optimizer_fn = optimizer
39
- else:
40
- raise TypeError(f"Invalid optimizer type: {type(optimizer)}")
41
- return optimizer_fn
42
-
43
-
44
- def get_scheduler(
45
- scheduler: (
46
- str
47
- | torch.optim.lr_scheduler._LRScheduler
48
- | torch.optim.lr_scheduler.LRScheduler
49
- | type[torch.optim.lr_scheduler._LRScheduler]
50
- | type[torch.optim.lr_scheduler.LRScheduler]
51
- | None
52
- ),
53
- optimizer,
54
- **scheduler_params,
55
- ):
56
- if isinstance(scheduler, str):
57
- if scheduler == "step":
58
- scheduler_fn = torch.optim.lr_scheduler.StepLR(
59
- optimizer, **scheduler_params
60
- )
61
- elif scheduler == "cosine":
62
- scheduler_fn = torch.optim.lr_scheduler.CosineAnnealingLR(
63
- optimizer, **scheduler_params
64
- )
65
- else:
66
- raise NotImplementedError(f"Unsupported scheduler: {scheduler}")
67
- elif isinstance(
68
- scheduler,
69
- (torch.optim.lr_scheduler._LRScheduler, torch.optim.lr_scheduler.LRScheduler),
70
- ):
71
- scheduler_fn = scheduler
72
- else:
73
- raise TypeError(f"Invalid scheduler type: {type(scheduler)}")
74
-
75
- return scheduler_fn
nextrec/utils/tensor.py DELETED
@@ -1,72 +0,0 @@
1
- """
2
- Tensor manipulation utilities for NextRec
3
-
4
- Date: create on 03/12/2025
5
- Author: Yang Zhou, zyaztec@gmail.com
6
- """
7
-
8
- import torch
9
- from typing import Any
10
-
11
-
12
- def to_tensor(
13
- value: Any, dtype: torch.dtype, device: torch.device | str | None = None
14
- ) -> torch.Tensor:
15
- if value is None:
16
- raise ValueError("[Tensor Utils Error] Cannot convert None to tensor.")
17
- tensor = value if isinstance(value, torch.Tensor) else torch.as_tensor(value)
18
- if tensor.dtype != dtype:
19
- tensor = tensor.to(dtype=dtype)
20
-
21
- if device is not None:
22
- target_device = (
23
- device if isinstance(device, torch.device) else torch.device(device)
24
- )
25
- if tensor.device != target_device:
26
- tensor = tensor.to(target_device)
27
- return tensor
28
-
29
-
30
- def stack_tensors(tensors: list[torch.Tensor], dim: int = 0) -> torch.Tensor:
31
- if not tensors:
32
- raise ValueError("[Tensor Utils Error] Cannot stack empty list of tensors.")
33
- return torch.stack(tensors, dim=dim)
34
-
35
-
36
- def concat_tensors(tensors: list[torch.Tensor], dim: int = 0) -> torch.Tensor:
37
- if not tensors:
38
- raise ValueError(
39
- "[Tensor Utils Error] Cannot concatenate empty list of tensors."
40
- )
41
- return torch.cat(tensors, dim=dim)
42
-
43
-
44
- def pad_sequence_tensors(
45
- tensors: list[torch.Tensor],
46
- max_len: int | None = None,
47
- padding_value: float = 0.0,
48
- padding_side: str = "right",
49
- ) -> torch.Tensor:
50
- if not tensors:
51
- raise ValueError("[Tensor Utils Error] Cannot pad empty list of tensors.")
52
- if max_len is None:
53
- max_len = max(t.size(0) for t in tensors)
54
- batch_size = len(tensors)
55
- padded = torch.full(
56
- (batch_size, max_len),
57
- padding_value,
58
- dtype=tensors[0].dtype,
59
- device=tensors[0].device,
60
- )
61
-
62
- for i, tensor in enumerate(tensors):
63
- length = min(tensor.size(0), max_len)
64
- if padding_side == "right":
65
- padded[i, :length] = tensor[:length]
66
- elif padding_side == "left":
67
- padded[i, -length:] = tensor[:length]
68
- else:
69
- raise ValueError(
70
- f"[Tensor Utils Error] padding_side must be 'right' or 'left', got {padding_side}"
71
- )
72
- return padded
@@ -1,71 +0,0 @@
1
- nextrec/__init__.py,sha256=_M3oUqyuvQ5k8Th_3wId6hQ_caclh7M5ad51XN09m98,235
2
- nextrec/__version__.py,sha256=40-PUZPRIakJU2yYWQcwTYvSJA6iewqiG8XylhxuAQk,22
3
- nextrec/cli.py,sha256=ItikUEFDju7c_3tDhYQuNPbhvFK6jij9zAujR_MYYK8,19388
4
- nextrec/basic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
5
- nextrec/basic/activation.py,sha256=uzTWfCOtBSkbu_Gk9XBNTj8__s241CaYLJk6l8nGX9I,2885
6
- nextrec/basic/callback.py,sha256=lELZhkB1TlAT_5My7Ce41u7Rox73rbyaulykWCk-svo,14108
7
- nextrec/basic/features.py,sha256=kfgU3gCORjYD7ylmKJDrWIrRIL97GcKWCDWeULU5N7g,4339
8
- nextrec/basic/layers.py,sha256=jDT6nnpCxdgzKMdhU-apxPgjHm-AHujRxjgBum4mBm0,29017
9
- nextrec/basic/loggers.py,sha256=p9wNmLuRYyvHsOzP0eNOYSlV3hrTDjrt6ggrH_r4RE0,6243
10
- nextrec/basic/metrics.py,sha256=EicHCyyr2j45J-emHf5KifQN6fnb6IV6gqEBu7ioANQ,26066
11
- nextrec/basic/model.py,sha256=tYpQAbeW99aXTrsmryoWe9sA8ihG8pHNytRcU_49-pQ,100764
12
- nextrec/basic/session.py,sha256=UOG_-EgCOxvqZwCkiEd8sgNV2G1sm_HbzKYVQw8yYDI,4483
13
- nextrec/data/__init__.py,sha256=auT_PkbgU9pUCt7KQl6H2ajcUorRhSyHa8NG3wExcG8,1197
14
- nextrec/data/batch_utils.py,sha256=FAJiweuDyAIzX7rICVmcxMofdFs2-7RLinovwB-lAYM,2878
15
- nextrec/data/data_processing.py,sha256=lnpJWqa5KwxxtlDPJ5PsVtO9uKke0u4OaJJvgudjRXI,5302
16
- nextrec/data/data_utils.py,sha256=LaVNXATcqu0ARPV-6WESQz6JXi3g-zq4uKjcoqBFlqI,1219
17
- nextrec/data/dataloader.py,sha256=L4VBpWUZrxozFBV54nhJAAC-ZX5Hg6zFwIwpGnguJ9c,18789
18
- nextrec/data/preprocessor.py,sha256=BxoD6GHEre86i-TbxPi58Uwmg_G7oLkiER6f7VfmVHo,41583
19
- nextrec/loss/__init__.py,sha256=mO5t417BneZ8Ysa51GyjDaffjWyjzFgPXIQrrggasaQ,827
20
- nextrec/loss/listwise.py,sha256=UT9vJCOTOQLogVwaeTV7Z5uxIYnngGdxk-p9e97MGkU,5744
21
- nextrec/loss/loss_utils.py,sha256=8lGJpiJ7GYplUdQqJL5XzHIiQr98g5pRb2cTYdS5VLA,4646
22
- nextrec/loss/pairwise.py,sha256=X9yg-8pcPt2IWU0AiUhWAt3_4W_3wIF0uSdDYTdoPFY,3398
23
- nextrec/loss/pointwise.py,sha256=o9J3OznY0hlbDsUXqn3k-BBzYiuUH5dopz8QBFqS_kQ,7343
24
- nextrec/models/generative/__init__.py,sha256=RU4DRibbfK802rzSXIgf3mSwYsSsYLKJiPzcXqprbvY,350
25
- nextrec/models/generative/hstu.py,sha256=ro0ifzvlS0SQaS1PMVDjvGDBgql5RnWEPGCu_iRtMrQ,19605
26
- nextrec/models/generative/rqvae.py,sha256=yxZ9KNEjRu0yxBjeNtCa2Y0ElB97diV_Z92CS9hbhzE,29255
27
- nextrec/models/generative/tiger.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
28
- nextrec/models/match/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
29
- nextrec/models/match/dssm.py,sha256=W6oBOEMGT-xYvhd-qzmGpYqxm9B-fwIQaXE7nEw2gfI,7923
30
- nextrec/models/match/dssm_v2.py,sha256=I2eWWyzFd7nIiShOB7VPxNMLoqST5IY98WxjDyvAGa8,6955
31
- nextrec/models/match/mind.py,sha256=semg4EEThD4l7K8fW1wuRF3i9Ex-smJVI4CNORVajvA,15089
32
- nextrec/models/match/sdm.py,sha256=kwDjDkFpoTmicfKScXjhJLe500upbg8eDXICQLMW03k,10557
33
- nextrec/models/match/youtube_dnn.py,sha256=-XuaM7lh02qkLCwJF3KgWxOQGn61p92MnYA100LWG4o,7283
34
- nextrec/models/multi_task/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
35
- nextrec/models/multi_task/esmm.py,sha256=tQg_jE51VDTyc-F0auviyP8CI9uzYQ_KjybbCAXWp1s,6491
36
- nextrec/models/multi_task/mmoe.py,sha256=qFWKdCE_VSGpVrMgx0NOO-HtLRNGdVxCWdkMfoEgjLA,8583
37
- nextrec/models/multi_task/ple.py,sha256=SMTgKqz8huXzmyMwACVG8yisHvd3GFGshYl7LOpnJXs,13016
38
- nextrec/models/multi_task/poso.py,sha256=JkNlMcqjMuE4PTGM6HeGcJTxhbLklXpusfyY8A1BjTQ,19017
39
- nextrec/models/multi_task/share_bottom.py,sha256=mkWaGHimUqp-2dmPHXjb5ffxX7ixv1BF0gQXTbx9kBo,6519
40
- nextrec/models/ranking/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
41
- nextrec/models/ranking/afm.py,sha256=XaiUYm36-pVNzB31lEtMstjg42-shn94khja0LMQB3s,10125
42
- nextrec/models/ranking/autoint.py,sha256=CyHnYyHJiQIOiPGI-j_16nCpECDQJ3FlVZ4nq3qu-l8,8109
43
- nextrec/models/ranking/dcn.py,sha256=vxbrDu9RxXznXNpXVeYJR4wdxoc4Vo0ygML6fFArY18,7299
44
- nextrec/models/ranking/dcn_v2.py,sha256=VNMiHf6BeBOxnoommjGZfF-9t_B88niiVEgmPVcGjQM,11163
45
- nextrec/models/ranking/deepfm.py,sha256=D9RPM40QAhogw8_RAOfE3JD1gnGf4F3-gXR40EZq-RU,5224
46
- nextrec/models/ranking/dien.py,sha256=G1W_pj8XyGBPgZo_86I3LgfHzQvR-xvR-PGNJZdRdAM,18958
47
- nextrec/models/ranking/din.py,sha256=gcibKTxK6nQCCxYMymO9ttu3UG2MSrOWRNBPCmJgMEM,9422
48
- nextrec/models/ranking/fibinet.py,sha256=OuE4MoG7rHycyRRQtKOvxHbuf7C6zoJFxGFerXmmn9U,7919
49
- nextrec/models/ranking/fm.py,sha256=ko_Eao9UfklakEk_TVEFZSyVAojmtclo1uIMBhL4FLU,4525
50
- nextrec/models/ranking/masknet.py,sha256=JqBSW1Wh_-P2OoEPss4FBhFdsBfTJYBsj9RR0Eoy6pk,12373
51
- nextrec/models/ranking/pnn.py,sha256=twwixy26mfAVaI9AqNnMLdwOG-WtDga60xsNiyJrFjI,8174
52
- nextrec/models/ranking/widedeep.py,sha256=Xm2klmKBOoSKWCBQN7FhwLStu0BHSTOgAJ9kwLmtiFY,5077
53
- nextrec/models/ranking/xdeepfm.py,sha256=kcPLoNC1940YxRMgWZS4mSxIXlwtc_HfNDIae_uYrsU,8156
54
- nextrec/utils/__init__.py,sha256=zqU9vjRUpVzJepcvdbxboik68K5jnMR40kdVjr6tpXY,2599
55
- nextrec/utils/cli_utils.py,sha256=mOI2yKZwCE3lVH5RCKs3WnIKyXYiLI08vHFT8duBRrQ,1497
56
- nextrec/utils/config.py,sha256=wx2Y-oxQqWhwOlW6DdG6ExLXDf-QDRnj8EQYDjrVhuY,19903
57
- nextrec/utils/device.py,sha256=DtgmrJnVJQKtgtVUbm0SW0vZ5Le0R9HU8TsvqPnRLZc,2453
58
- nextrec/utils/distributed.py,sha256=tIkgUjzEjR_FHOm9ckyM8KddkCfxNSogP-rdHcVGhuk,4782
59
- nextrec/utils/embedding.py,sha256=akAEc062MG2cD7VIOllHaqtwzAirQR2gq5iW7oKpGAU,1449
60
- nextrec/utils/feature.py,sha256=LcXaWP98zMZhJTKL92VVHX8mqOE5Q0MyVq3hw5Z9kxs,300
61
- nextrec/utils/file.py,sha256=s2cO1LRbU7xPeAbVoOA6XOoV6wvLrW6oy6p9fVSz9pc,3024
62
- nextrec/utils/initializer.py,sha256=sEkP5DUQ4x2Cydc2h5MoGzBhLr-4tt6x6bqTO2f0TyE,2206
63
- nextrec/utils/model.py,sha256=dYl1XfIZt6aVjNyV2AAhcArwFRMcEAKrjG_pr8AVHs0,1163
64
- nextrec/utils/optimizer.py,sha256=eX8baIvWOpwDTGninbyp6pQfzdHbIL62GTi4ldpYcfM,2337
65
- nextrec/utils/synthetic_data.py,sha256=V6lxEQCU-yJXMBCoCAeKZZWBMYXpZhvbyfiI8jCmvqs,18112
66
- nextrec/utils/tensor.py,sha256=Z6MBpSuQpHw4kGjeKxG0cXZMpRBCM45zTKhk9WolyiM,2220
67
- nextrec-0.4.8.dist-info/METADATA,sha256=25dlzG2TNFZCvm_V6ZAnqaMTL6qwNNt7xMSRsmaQ_Nk,19460
68
- nextrec-0.4.8.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
69
- nextrec-0.4.8.dist-info/entry_points.txt,sha256=NN-dNSdfMRTv86bNXM7d3ZEPW2BQC6bRi7QP7i9cIps,45
70
- nextrec-0.4.8.dist-info/licenses/LICENSE,sha256=2fQfVKeafywkni7MYHyClC6RGGC3laLTXCNBx-ubtp0,1064
71
- nextrec-0.4.8.dist-info/RECORD,,
File without changes