replay-rec 0.18.0rc0__py3-none-any.whl → 0.18.1__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 (90) hide show
  1. replay/__init__.py +1 -1
  2. replay/data/dataset.py +27 -1
  3. replay/data/dataset_utils/dataset_label_encoder.py +6 -3
  4. replay/data/nn/schema.py +37 -16
  5. replay/data/nn/sequence_tokenizer.py +313 -165
  6. replay/data/nn/torch_sequential_dataset.py +17 -8
  7. replay/data/nn/utils.py +14 -7
  8. replay/data/schema.py +10 -6
  9. replay/metrics/offline_metrics.py +2 -2
  10. replay/models/__init__.py +1 -0
  11. replay/models/base_rec.py +18 -21
  12. replay/models/lin_ucb.py +407 -0
  13. replay/models/nn/sequential/bert4rec/dataset.py +17 -4
  14. replay/models/nn/sequential/bert4rec/lightning.py +121 -54
  15. replay/models/nn/sequential/bert4rec/model.py +21 -0
  16. replay/models/nn/sequential/callbacks/prediction_callbacks.py +5 -1
  17. replay/models/nn/sequential/compiled/__init__.py +5 -0
  18. replay/models/nn/sequential/compiled/base_compiled_model.py +261 -0
  19. replay/models/nn/sequential/compiled/bert4rec_compiled.py +152 -0
  20. replay/models/nn/sequential/compiled/sasrec_compiled.py +145 -0
  21. replay/models/nn/sequential/postprocessors/postprocessors.py +27 -1
  22. replay/models/nn/sequential/sasrec/dataset.py +17 -1
  23. replay/models/nn/sequential/sasrec/lightning.py +126 -50
  24. replay/models/nn/sequential/sasrec/model.py +3 -4
  25. replay/preprocessing/__init__.py +7 -1
  26. replay/preprocessing/discretizer.py +719 -0
  27. replay/preprocessing/label_encoder.py +384 -52
  28. replay/splitters/cold_user_random_splitter.py +1 -1
  29. replay/utils/__init__.py +1 -0
  30. replay/utils/common.py +7 -8
  31. replay/utils/session_handler.py +3 -4
  32. replay/utils/spark_utils.py +15 -1
  33. replay/utils/types.py +8 -0
  34. {replay_rec-0.18.0rc0.dist-info → replay_rec-0.18.1.dist-info}/METADATA +75 -70
  35. {replay_rec-0.18.0rc0.dist-info → replay_rec-0.18.1.dist-info}/RECORD +37 -84
  36. {replay_rec-0.18.0rc0.dist-info → replay_rec-0.18.1.dist-info}/WHEEL +1 -1
  37. replay/experimental/__init__.py +0 -0
  38. replay/experimental/metrics/__init__.py +0 -62
  39. replay/experimental/metrics/base_metric.py +0 -602
  40. replay/experimental/metrics/coverage.py +0 -97
  41. replay/experimental/metrics/experiment.py +0 -175
  42. replay/experimental/metrics/hitrate.py +0 -26
  43. replay/experimental/metrics/map.py +0 -30
  44. replay/experimental/metrics/mrr.py +0 -18
  45. replay/experimental/metrics/ncis_precision.py +0 -31
  46. replay/experimental/metrics/ndcg.py +0 -49
  47. replay/experimental/metrics/precision.py +0 -22
  48. replay/experimental/metrics/recall.py +0 -25
  49. replay/experimental/metrics/rocauc.py +0 -49
  50. replay/experimental/metrics/surprisal.py +0 -90
  51. replay/experimental/metrics/unexpectedness.py +0 -76
  52. replay/experimental/models/__init__.py +0 -10
  53. replay/experimental/models/admm_slim.py +0 -205
  54. replay/experimental/models/base_neighbour_rec.py +0 -204
  55. replay/experimental/models/base_rec.py +0 -1271
  56. replay/experimental/models/base_torch_rec.py +0 -234
  57. replay/experimental/models/cql.py +0 -454
  58. replay/experimental/models/ddpg.py +0 -923
  59. replay/experimental/models/dt4rec/__init__.py +0 -0
  60. replay/experimental/models/dt4rec/dt4rec.py +0 -189
  61. replay/experimental/models/dt4rec/gpt1.py +0 -401
  62. replay/experimental/models/dt4rec/trainer.py +0 -127
  63. replay/experimental/models/dt4rec/utils.py +0 -265
  64. replay/experimental/models/extensions/spark_custom_models/__init__.py +0 -0
  65. replay/experimental/models/extensions/spark_custom_models/als_extension.py +0 -792
  66. replay/experimental/models/implicit_wrap.py +0 -131
  67. replay/experimental/models/lightfm_wrap.py +0 -302
  68. replay/experimental/models/mult_vae.py +0 -332
  69. replay/experimental/models/neuromf.py +0 -406
  70. replay/experimental/models/scala_als.py +0 -296
  71. replay/experimental/nn/data/__init__.py +0 -1
  72. replay/experimental/nn/data/schema_builder.py +0 -55
  73. replay/experimental/preprocessing/__init__.py +0 -3
  74. replay/experimental/preprocessing/data_preparator.py +0 -839
  75. replay/experimental/preprocessing/padder.py +0 -229
  76. replay/experimental/preprocessing/sequence_generator.py +0 -208
  77. replay/experimental/scenarios/__init__.py +0 -1
  78. replay/experimental/scenarios/obp_wrapper/__init__.py +0 -8
  79. replay/experimental/scenarios/obp_wrapper/obp_optuna_objective.py +0 -74
  80. replay/experimental/scenarios/obp_wrapper/replay_offline.py +0 -248
  81. replay/experimental/scenarios/obp_wrapper/utils.py +0 -87
  82. replay/experimental/scenarios/two_stages/__init__.py +0 -0
  83. replay/experimental/scenarios/two_stages/reranker.py +0 -117
  84. replay/experimental/scenarios/two_stages/two_stages_scenario.py +0 -757
  85. replay/experimental/utils/__init__.py +0 -0
  86. replay/experimental/utils/logger.py +0 -24
  87. replay/experimental/utils/model_handler.py +0 -186
  88. replay/experimental/utils/session_handler.py +0 -44
  89. replay_rec-0.18.0rc0.dist-info/NOTICE +0 -41
  90. {replay_rec-0.18.0rc0.dist-info → replay_rec-0.18.1.dist-info}/LICENSE +0 -0
replay/models/nn/sequential/sasrec/model.py CHANGED
@@ -57,8 +57,8 @@ class SasRecModel(torch.nn.Module):
57
57
  self.padding_idx = item_count
58
58
 
59
59
  assert schema.item_id_feature_name
60
+ self.schema = schema
60
61
  self.item_feature_name = schema.item_id_feature_name
61
- self.register_buffer("candidates_to_score", torch.LongTensor(list(range(self.item_count))))
62
62
 
63
63
  # Model blocks
64
64
  self.masking = SasRecMasks(
@@ -205,7 +205,6 @@ class SasRecMasks:
205
205
  :param padding_idx: Padding indices.
206
206
  """
207
207
  assert schema.item_id_feature_name
208
- self.schema = schema
209
208
  self.item_feature_name = schema.item_id_feature_name
210
209
  self.padding_idx = padding_idx
211
210
 
@@ -223,8 +222,8 @@ class SasRecMasks:
223
222
  input_sequence = feature_tensor[self.item_feature_name]
224
223
 
225
224
  attention_mask = ~torch.tril(
226
- torch.ones((input_sequence.shape[1], input_sequence.shape[1]), dtype=torch.bool)
227
- ).to(padding_mask.device)
225
+ torch.ones((input_sequence.shape[1], input_sequence.shape[1]), dtype=torch.bool, device=padding_mask.device)
226
+ )
228
227
 
229
228
  output_feature_tensor = dict(feature_tensor)
230
229
  output_feature_tensor[self.item_feature_name] = input_sequence.masked_fill(
replay/preprocessing/__init__.py CHANGED
@@ -6,11 +6,17 @@ This module contains tools for preprocessing data including:
6
6
  """
7
7
 
8
8
  from .converter import CSRConverter
9
+ from .discretizer import (
10
+ Discretizer,
11
+ GreedyDiscretizingRule,
12
+ HandleInvalidStrategies,
13
+ QuantileDiscretizingRule,
14
+ )
9
15
  from .history_based_fp import (
10
16
  ConditionalPopularityProcessor,
11
17
  EmptyFeatureProcessor,
12
18
  HistoryBasedFeaturesProcessor,
13
19
  LogStatFeaturesProcessor,
14
20
  )
15
- from .label_encoder import LabelEncoder, LabelEncodingRule
21
+ from .label_encoder import LabelEncoder, LabelEncoderPartialFitWarning, LabelEncodingRule, SequenceEncodingRule
16
22
  from .sessionizer import Sessionizer