replay-rec 0.16.0__py3-none-any.whl → 0.17.0__py3-none-any.whl

replay/utils/time.py

@@ -1,7 +1,6 @@
 import numpy as np
 
 from .spark_utils import convert2spark
-
 from .types import PYSPARK_AVAILABLE, DataFrameLike
 
 if PYSPARK_AVAILABLE:
@@ -106,9 +105,7 @@ def get_item_recency(
         "item_idx",
         sf.unix_timestamp(sf.to_timestamp("timestamp")).alias("timestamp"),
     )
-    items = items.groupBy("item_idx").agg(
-        sf.mean("timestamp").alias("timestamp")
-    )
+    items = items.groupBy("item_idx").agg(sf.mean("timestamp").alias("timestamp"))
     items = items.withColumn("relevance", sf.lit(1))
     items = smoothe_time(items, decay, limit, kind)
     return items
@@ -234,16 +231,10 @@ def smoothe_time(
     <BLANKLINE>
     """
     log = convert2spark(log)
-    log = log.withColumn(
-        "timestamp", sf.unix_timestamp(sf.to_timestamp("timestamp"))
-    )
-    last_date = (
-        log.agg({"timestamp": "max"}).collect()[0].asDict()["max(timestamp)"]
-    )
+    log = log.withColumn("timestamp", sf.unix_timestamp(sf.to_timestamp("timestamp")))
+    last_date = log.agg({"timestamp": "max"}).collect()[0].asDict()["max(timestamp)"]
     day_in_secs = 86400
-    log = log.withColumn(
-        "age", (last_date - sf.col("timestamp")) / day_in_secs
-    )
+    log = log.withColumn("age", (last_date - sf.col("timestamp")) / day_in_secs)
     if kind == "power":
         power = np.log(0.5) / np.log(decay)
         log = log.withColumn("age", sf.pow(sf.col("age") + 1, power))
@@ -254,15 +245,10 @@ def smoothe_time(
         k = 0.5 / decay
         log = log.withColumn("age", 1 - k * sf.col("age"))
     else:
-        raise ValueError(
-            f"parameter kind must be one of [power, exp, linear], got {kind}"
-        )
+        msg = f"parameter kind must be one of [power, exp, linear], got {kind}"
+        raise ValueError(msg)
 
-    log = log.withColumn(
-        "age", sf.when(sf.col("age") < limit, limit).otherwise(sf.col("age"))
-    )
-    log = log.withColumn(
-        "relevance", sf.col("relevance") * sf.col("age")
-    ).drop("age")
+    log = log.withColumn("age", sf.when(sf.col("age") < limit, limit).otherwise(sf.col("age")))
+    log = log.withColumn("relevance", sf.col("relevance") * sf.col("age")).drop("age")
     log = log.withColumn("timestamp", sf.to_timestamp("timestamp"))
     return log

replay/utils/types.py

@@ -4,7 +4,6 @@ from pandas import DataFrame as PandasDataFrame
 from polars import DataFrame as PolarsDataFrame
 
 
-# pylint: disable=too-few-public-methods
 class MissingImportType:
     """
     Replacement class with missing import
@@ -17,10 +16,9 @@ try:
     PYSPARK_AVAILABLE = True
 except ImportError:
     PYSPARK_AVAILABLE = False
-    SparkDataFrame = MissingImportType  # type: ignore
+    SparkDataFrame = MissingImportType
 
 try:
-    # pylint: disable=unused-import
     import torch  # noqa: F401
 
     TORCH_AVAILABLE = True

{replay_rec-0.16.0.dist-info → replay_rec-0.17.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: replay-rec
-Version: 0.16.0
+Version: 0.17.0
 Summary: RecSys Library
 Home-page: https://sb-ai-lab.github.io/RePlay/
 License: Apache-2.0
@@ -28,7 +28,7 @@ Requires-Dist: optuna (>=3.2.0,<3.3.0)
 Requires-Dist: pandas (>=1.3.5,<2.0.0)
 Requires-Dist: polars (>=0.20.7,<0.21.0)
 Requires-Dist: psutil (>=5.9.5,<5.10.0)
-Requires-Dist: pyarrow (>=12.0.1,<=14.0.1)
+Requires-Dist: pyarrow (>=12.0.1)
 Requires-Dist: pyspark (>=3.0,<3.3) ; extra == "spark" or extra == "all"
 Requires-Dist: pytorch-ranger (>=0.1.1,<0.2.0) ; extra == "torch" or extra == "all"
 Requires-Dist: scikit-learn (>=1.0.2,<2.0.0)

replay_rec-0.17.0.dist-info/RECORD

@@ -0,0 +1,127 @@
+replay/__init__.py,sha256=knHFGIzZuD7qKkGHgDbbA_GnYg1omuXxLDS4JVvYEdk,46
+replay/data/__init__.py,sha256=g5bKRyF76QL_BqlED-31RnS8pBdcyj9loMsx5vAG_0E,301
+replay/data/dataset.py,sha256=ysMTNfx8I2hI9fSugtt3IPhenmutgzQMw-8VcM3oUJk,21299
+replay/data/dataset_utils/__init__.py,sha256=9wUvG8ZwGUvuzLU4zQI5FDcH0WVVo5YLN2ey3DterP0,55
+replay/data/dataset_utils/dataset_label_encoder.py,sha256=TEx2zLw5rJdIz1SRBEznyVv5x_Cs7o6QQbzMk-M1LU0,9598
+replay/data/nn/__init__.py,sha256=WxLsi4rgOuuvGYHN49xBPxP2Srhqf3NYgfBDVH-ZvBo,1122
+replay/data/nn/schema.py,sha256=BYU65vLqPDl69OE-rReh59fiQK0ERfs1xbBLWCiIJnw,14258
+replay/data/nn/sequence_tokenizer.py,sha256=dXD8l7IfK1dod8p--I6BhvE9af3iUOfpaoW2QBU9hTs,34133
+replay/data/nn/sequential_dataset.py,sha256=fqlyBAzDmpH332S-LoMP9PoRYMtgZpxG6Qdahmk5GtE,7840
+replay/data/nn/torch_sequential_dataset.py,sha256=BqrK_PtkhpsaY1zRIWGk4EgwPL31a7IWCc0hLDuwDQc,10984
+replay/data/nn/utils.py,sha256=YKE9gkIHZDDiwv4THqOWL4PzsdOujnPuM97v79Mwq0E,2769
+replay/data/schema.py,sha256=F_cv6sYb6l23yuX5xWnbqoJ9oSeUT2NpIM19u8Lf2jA,15606
+replay/data/spark_schema.py,sha256=4o0Kn_fjwz2-9dBY3q46F9PL0F3E7jdVpIlX7SG3OZI,1111
+replay/metrics/__init__.py,sha256=KDkxVnKa4ks9K9GmlrdTx1pkIl-MAmm78ZASsp2ZndE,2812
+replay/metrics/base_metric.py,sha256=uleW5vLrdA3iRx72tFyW0cxe6ne_ugQ1XaY_ZTcnAOo,15960
+replay/metrics/categorical_diversity.py,sha256=OYsF-Ng-WrF9CC-sKgQKngrA779NO8MtgRvvAyC8MXM,10781
+replay/metrics/coverage.py,sha256=wE1Y_TgKOzf_9ixeas-vsxANAHeHSGPuGrzKk8DklaY,8843
+replay/metrics/descriptors.py,sha256=BHORyGKfJgPeUjgLO0u2urSTe16UQbb-HHh8soqnwDE,3893
+replay/metrics/experiment.py,sha256=Ab43e-eyoy8RM_OgAcm0toqFNDaWkh5Gbq_XVxbmqB0,8142
+replay/metrics/hitrate.py,sha256=G0_GufGc1RIQ_RntCIvDn0piJIVyT3W35bTN-Un9-Ug,2339
+replay/metrics/map.py,sha256=qUPFkyrvRf7gEyxosBApHOxpAVZJ3Yksqe4vsbzTukE,2597
+replay/metrics/mrr.py,sha256=AkfRdvL21Ri1zE4Cmib2twJaEaeBO5wze87eZ4lkYlc,2188
+replay/metrics/ndcg.py,sha256=wzdFFf9rmG6yH7klK5V9ab1wNdI6CHXezvYasxO9_ZU,3134
+replay/metrics/novelty.py,sha256=4S6PsdOl8lqweBlemWJh2fVqMiN8jfj3cIU-9TxZXLY,5500
+replay/metrics/offline_metrics.py,sha256=bVI47C-oTxyi-ksTEdTckX3fT_Dz7koi0ICTgUmsnRM,20468
+replay/metrics/precision.py,sha256=8hfz9bLZt7wUfyJtJPiIoDEOP_bweZSpCS6Z1Cm9IGE,2281
+replay/metrics/recall.py,sha256=ATN5OIeFKMG7-nqpPCBPLItrUhGRucLCELWnzRyCRoo,2472
+replay/metrics/rocauc.py,sha256=gybFcyIyVDpwoR9K38nOSxlAjIYbFqXUzABuckxqHps,3307
+replay/metrics/surprisal.py,sha256=wj9Q5mAdECpl0LfykJWt8jgN3_CUSlai2fhiFgJr_Vw,7474
+replay/metrics/torch_metrics_builder.py,sha256=2gcCcb0A-TVpYcBIYGhXrggyFX-M_T7Q1pQUiMpxEZE,13845
+replay/metrics/unexpectedness.py,sha256=cfDnkpK6nPeawwHDVNQAkUtsW0SvAttI84k4M5ttkyo,6888
+replay/models/__init__.py,sha256=_4gNsauyrVMYEoFDihPYY9kGuBGGFyy1krvxF7oEYjk,808
+replay/models/als.py,sha256=dpBwyg1ZBqtdgrFluHaq5nuPQT---fmA-N2TspJAM0U,6232
+replay/models/association_rules.py,sha256=cp4myXvMqro6zLMjJzJMb0DZ5DQFQEZvhqf5OBgBw8Y,14659
+replay/models/base_neighbour_rec.py,sha256=zMORSm4uMQSNj12v0n_6w8fVHgSYjeiqyYE9rrWgSfU,7887
+replay/models/base_rec.py,sha256=iF0eMlNQVcd-nb3aCRG3ObpmEi7P4-jP_5mKjwc6anc,66407
+replay/models/cat_pop_rec.py,sha256=tzI1UMlC3kEOrtDZ1UPpCP13tX8CeDJP7PHwQKl9Mmo,11922
+replay/models/cluster.py,sha256=9JcpGnbfgFa4UsyxPAa4WMuJFa3rsuAxiKoy-s_UfyE,4970
+replay/models/extensions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+replay/models/extensions/ann/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+replay/models/extensions/ann/ann_mixin.py,sha256=30YTIGnRhGJYOnxo2bM6dQFgDq9i5UH4t8Ln8zCW8TM,7288
+replay/models/extensions/ann/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+replay/models/extensions/ann/entities/base_hnsw_param.py,sha256=5GRdcQj4-zhNXfJ7ko2WHGHgRuXCzSHCRcRxljl1V4c,776
+replay/models/extensions/ann/entities/hnswlib_param.py,sha256=j3V4JXM_yfR6s2TjYakIXMg-zS1-MrP6an930DEIWGM,2104
+replay/models/extensions/ann/entities/nmslib_hnsw_param.py,sha256=WeEhRR4jKqgvWK_zDK8fx6kEqc2e_bc0kubvqK3iV8c,2162
+replay/models/extensions/ann/index_builders/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+replay/models/extensions/ann/index_builders/base_index_builder.py,sha256=Ul25G0FaNLOXUjrDXxZDTg7tLXlv1N6wR8kWjWICtZ0,2110
+replay/models/extensions/ann/index_builders/driver_hnswlib_index_builder.py,sha256=bUoDwelKL9OK4z0IABzDm16t8wQ4QzH3wQ5NP32S_IE,1588
+replay/models/extensions/ann/index_builders/driver_nmslib_index_builder.py,sha256=1NLWyAJGYgp46uUBhUYQyd0stmG6DhLh7U4JEne5TFw,1308
+replay/models/extensions/ann/index_builders/executor_hnswlib_index_builder.py,sha256=_JZfewcLsU3sRR95_FPlgLfn9vez0JXMxer_024UHK4,2449
+replay/models/extensions/ann/index_builders/executor_nmslib_index_builder.py,sha256=_TClG_czFb50dXyUjjXL4HE5j_i4BH011wjzDGba2U8,2591
+replay/models/extensions/ann/index_builders/nmslib_index_builder_mixin.py,sha256=AIkVnobesnTM5lrBSWf9gd0CySwFQ0vH_DjemfLS4Cs,1925
+replay/models/extensions/ann/index_inferers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+replay/models/extensions/ann/index_inferers/base_inferer.py,sha256=I39aqEc2somfndrCd-KC3XYZnYSrJ2hGpR9y6wO93NA,2524
+replay/models/extensions/ann/index_inferers/hnswlib_filter_index_inferer.py,sha256=JjT4l_XAjzUOsTAE7OS88zAgPd_h_O44oUnn2kVr8E0,2477
+replay/models/extensions/ann/index_inferers/hnswlib_index_inferer.py,sha256=CoY_oMfdcwnh87ceuSpHXu4Czle9xxeMisO8XJUuJLE,1717
+replay/models/extensions/ann/index_inferers/nmslib_filter_index_inferer.py,sha256=1bpBjRhj4J_ecaORRhkhEke7ImJcxVTFRmmGK2wISB4,3120
+replay/models/extensions/ann/index_inferers/nmslib_index_inferer.py,sha256=TqyunbjMQp1bWltbouvqK2kr2cnER6_d75NuCTVB3O0,2195
+replay/models/extensions/ann/index_inferers/utils.py,sha256=6IST2FPSY3nuYu5KqzRpd4FgdaV3GnQRQlxp9LN_yyA,641
+replay/models/extensions/ann/index_stores/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+replay/models/extensions/ann/index_stores/base_index_store.py,sha256=u4l2ybAXX92ZMGK7NqqosbKF75QgFqhAMVadd5ePj6Y,910
+replay/models/extensions/ann/index_stores/hdfs_index_store.py,sha256=0zDq9PdDOiD6HvtZlfjTbuJHfeTOWBTQ_HiuqZmoxtA,3090
+replay/models/extensions/ann/index_stores/shared_disk_index_store.py,sha256=thl4T1uYU4Gtk4nBao_qK8CbFBdX1xmXNishxgfCd-I,2030
+replay/models/extensions/ann/index_stores/spark_files_index_store.py,sha256=QP_8mE7EIBbePIe0AB-IWuJLRA5MR3wswCEt8oHzn-0,3617
+replay/models/extensions/ann/index_stores/utils.py,sha256=6r2GP_EFCaCguolW857pb4lRS8rh6_Nv_Edso9_j5no,3756
+replay/models/extensions/ann/utils.py,sha256=qLeT0wHzbzBU5J6SE3xbbT3vYrLrFcEEQNJCtElvP_U,1213
+replay/models/kl_ucb.py,sha256=L6vC2KsTBTTx4ckmGhWybOiLa5Wt54N7cgl7jS2FQRg,6731
+replay/models/knn.py,sha256=ujJRiRJ9Nfs_m7eTvk0fDTYiR95aKnDO4e2Tf2q2AQI,9109
+replay/models/nn/__init__.py,sha256=AT3o1qXaxUq4_QIGlcGuSs54ZpueOo-SbpZwuGI-6os,41
+replay/models/nn/optimizer_utils/__init__.py,sha256=8MHln7CW54oACVUFKdZLjAf4bY83GcOMXpmL87gTnhI,178
+replay/models/nn/optimizer_utils/optimizer_factory.py,sha256=EXqS_2ZxT7K_kFzcjgFpNhmzUoSftf_CwZIIHNHZBgU,2842
+replay/models/nn/sequential/__init__.py,sha256=CI2n0cxs_amqJrwBMq6n0Z_uBOu7CGXfagqvE4Jlmjw,128
+replay/models/nn/sequential/bert4rec/__init__.py,sha256=JfZqHOGxcvOkICl5cWmZbZhaKXpkIvua-Wj57VWWEhw,399
+replay/models/nn/sequential/bert4rec/dataset.py,sha256=sCnYGF-sQ1YlLq7vofQo2GIIlc59YlbUgmW7bHI6MPg,10324
+replay/models/nn/sequential/bert4rec/lightning.py,sha256=TqO0V-g0JA0D-L2t08AgAIQgBkDtLUgl4xqekSiDWJ4,22605
+replay/models/nn/sequential/bert4rec/model.py,sha256=tiAiKOUwk3iPPYWyWkfOF23IzfL1NbeaF-8kNt9uZlU,21303
+replay/models/nn/sequential/callbacks/__init__.py,sha256=Q7mSZ_RB6iyD7QZaBL_NJ0uh8cRfgxq7gtPHbkSyhoo,282
+replay/models/nn/sequential/callbacks/prediction_callbacks.py,sha256=H4MZ87_N0hCKtHbsTuN-Cq_SJ-n9TSkvv2okuGnwo3M,9045
+replay/models/nn/sequential/callbacks/validation_callback.py,sha256=6TNl3NN9oahK1J7DT44461xqBuUCblCsLzUi2svlhF4,5825
+replay/models/nn/sequential/postprocessors/__init__.py,sha256=89LGzkNHukcuC2-rfpiz7vmv1zyk6MNY-8zaXrvtn0M,164
+replay/models/nn/sequential/postprocessors/_base.py,sha256=Z-HuXM4WBQtrq9IeCL00jtdb_UopksInxVVK8lxD_hA,1136
+replay/models/nn/sequential/postprocessors/postprocessors.py,sha256=V32xMyNPztJ5kapUxYSAz9i_vsqJ6_cjpGblmQGPO6A,6781
+replay/models/nn/sequential/sasrec/__init__.py,sha256=c6130lRpPkcbuGgkM7slagBIgH7Uk5zUtSzFDEwAsik,250
+replay/models/nn/sequential/sasrec/dataset.py,sha256=ReGNc6t9jjXxMZJp0WqFj1jatJFHnWOrkK3W8lwBNIs,7036
+replay/models/nn/sequential/sasrec/lightning.py,sha256=DtLnNikTNvqroCzaVFw7u-QZpZdvwiYbCwJLE7FkHms,21397
+replay/models/nn/sequential/sasrec/model.py,sha256=DE9kaqlcL22v07kpi2IzIwZ4-3AXNBVTZCnfuTS5usg,27775
+replay/models/pop_rec.py,sha256=Ju9y2rU2vW_jFU9-W15fbbr5_ZzYGihSjSxsqKsAf0Q,4964
+replay/models/query_pop_rec.py,sha256=UNsHtf3eQpJom73ZmEO5us4guI4SnCLJYTfuUpRgqes,4086
+replay/models/random_rec.py,sha256=9SC012_X3sNzrAjDG1CPGhjisZb6gnv4VCW7yIMSNpk,8066
+replay/models/slim.py,sha256=LHQl91PAsiCA8wLYzyapW1DVmqzGaCwtydBvZKC5yo0,4723
+replay/models/thompson_sampling.py,sha256=gcjlVl1mPiEVt70y8frA762O-eCZzd3SVg1lnDRCEHk,1939
+replay/models/ucb.py,sha256=X98ulD8L3gWR3VA7rbQkXFqQyzWc-Nt12lp_gbLTfLQ,6964
+replay/models/wilson.py,sha256=o7aUWjq3648dAfgGBoWD5Gu-HzdyobPMaH2lzCLijiA,4558
+replay/models/word2vec.py,sha256=MgoRIS5vqW9cH1HKAGa2xsLLnTH6XC1EXk4Dzvn5lXA,9171
+replay/optimization/__init__.py,sha256=az6U10rF7X6rPRUUPwLyiM1WFNJ_6kl0imA5xLVWFLs,120
+replay/optimization/optuna_objective.py,sha256=Z-8X0_FT3BicVWj0UhxoLrvZAck3Dhn7jHDGo0i0hxA,7653
+replay/preprocessing/__init__.py,sha256=TtBysFqYeDy4kZAEnWEaNSwPvbffYdfMkEs71YG51fM,411
+replay/preprocessing/converter.py,sha256=DczqsVLrwFi6EFhK2HR8rGiIxGCwXeY7QNgWorjA41g,4390
+replay/preprocessing/filters.py,sha256=6MaO4IIyKNFP2AR94YA5iQUhQvuCRhAFfj0opI6o4-Q,33744
+replay/preprocessing/history_based_fp.py,sha256=tfgKJPKm53LSNqM6VmMXYsVrRDc-rP1Tbzn8s3mbziQ,18751
+replay/preprocessing/label_encoder.py,sha256=MLBavPD-dB644as0E9ZJSE9-8QxGCB_IHek1w3xtqDI,27040
+replay/preprocessing/sessionizer.py,sha256=G6i0K3FwqtweRxvcSYraJ-tBWAT2HnV-bWHHlIZJF-s,12217
+replay/scenarios/__init__.py,sha256=kw2wRkPPinw0IBA20D83XQ3xeSudk3KuYAAA1Wdr8xY,93
+replay/scenarios/fallback.py,sha256=EeBmIR-5igzKR2m55bQRFyhxTkpJez6ZkCW449n8hWs,7130
+replay/splitters/__init__.py,sha256=DnqVMelrzLwR8fGQgcWN_8FipGs8T4XGSPOMW-L_x2g,454
+replay/splitters/base_splitter.py,sha256=qWW8Sueu0BrYt0WIxMbzooAC4-jhEmyd6pMND_H_qB0,7751
+replay/splitters/cold_user_random_splitter.py,sha256=gVwBVdn_0IOaLGT_UzJoS9AMaPhelZy-FpC5JQS1PhA,4136
+replay/splitters/k_folds.py,sha256=WH02_DP18A2ae893ysonmfLPB56_i1ETllTAwaCYekg,6218
+replay/splitters/last_n_splitter.py,sha256=r9kdq2JPi508C9ywjwc68an-iq27KsigMfHWLz0YohE,15346
+replay/splitters/new_users_splitter.py,sha256=bv_QCPkL7KFxJIovAXQbP3Rlty3My48YNTqrj-2ucFQ,9167
+replay/splitters/random_splitter.py,sha256=mbOcxeF0B9WQ9OSxK8CHkPtO8UzKCZJt3rRyFhn-hyQ,2996
+replay/splitters/ratio_splitter.py,sha256=8zvuCn16Icc4ntQPKXJ5ArAWuJzCZ9NHZtgWctKyBVY,17519
+replay/splitters/time_splitter.py,sha256=iXhuafjBx7dWyJSy-TEVy1IUQBwMpA1gAiF4-GtRe2g,9031
+replay/splitters/two_stage_splitter.py,sha256=PWozxjjgjrVzdz6Sm9dcDTeH0bOA24reFzkk_N_TgbQ,17734
+replay/utils/__init__.py,sha256=vDJgOWq81fbBs-QO4ZDpdqR4KDyO1kMOOxBRi-5Gp7E,253
+replay/utils/common.py,sha256=6JxR5bFuTFTFWad36J5Zu8dFgpFXoof6VsVpF2sD7h8,1471
+replay/utils/dataframe_bucketizer.py,sha256=LipmBBQkdkLGroZpbP9i7qvTombLdMxo2dUUys1m5OY,3748
+replay/utils/distributions.py,sha256=kGGq2KzQZ-yhTuw_vtOsKFXVpXUOQ2l4aIFBcaDufZ8,1202
+replay/utils/model_handler.py,sha256=V-mHDh8_UexjVSsMBBRA9yrjS_5MPHwYOwv_UrI-Zfs,6466
+replay/utils/session_handler.py,sha256=ijTvDSNAe1D9R1e-dhtd-r80tFNiIBsFdWZLgw-gLEo,5153
+replay/utils/spark_utils.py,sha256=PhNi9fW28ek0ZB90AUg3tsT5BULbQjDhLalxxww9eLE,26700
+replay/utils/time.py,sha256=J8asoQBytPcNw-BLGADYIsKeWhIoN1H5hKiX9t2AMqo,9376
+replay/utils/types.py,sha256=5sw0A7NG4ZgQKdWORnBy0wBZ5F98sP_Ju8SKQ6zbDS4,651
+replay_rec-0.17.0.dist-info/LICENSE,sha256=rPmcA7UrHxBChEAAlJyE24qUWKKl9yLQXxFsKeg_LX4,11344
+replay_rec-0.17.0.dist-info/METADATA,sha256=5wmgvIeFw22ST7-kZrWT9ggEAmne8I7_yHkwN1Bfus8,10588
+replay_rec-0.17.0.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+replay_rec-0.17.0.dist-info/RECORD,,

replay_rec-0.16.0.dist-info/RECORD

@@ -1,126 +0,0 @@
-replay/__init__.py,sha256=5QiEYYR9P626qiWyk47L6vQBjKSVroEKMYKGcH2HGt4,46
-replay/data/__init__.py,sha256=o8IzviPmnGQAoXGMgVoJRWJ-5WJ7ZjxeC9oc0cmttdU,301
-replay/data/dataset.py,sha256=ViDt66upex_kKKJ9a7F-WS5f6s4mbAa6bCIC3OHqCsY,21569
-replay/data/dataset_utils/__init__.py,sha256=9wUvG8ZwGUvuzLU4zQI5FDcH0WVVo5YLN2ey3DterP0,55
-replay/data/dataset_utils/dataset_label_encoder.py,sha256=8UZFWtia343vANf4VmQqoI8ZwLuqwYEY5ESWxFsB1Lw,9580
-replay/data/nn/__init__.py,sha256=GrUxoZucL5sGY_mZzxy6Vr_WIwSXeiD1k-HaBWQDnx4,1122
-replay/data/nn/schema.py,sha256=YzOM4hQ6c4mIHL-C50Nnh02GzhakMSG7NrClqURJYGo,14799
-replay/data/nn/sequence_tokenizer.py,sha256=ZXM0P5BJDomPX4lN4vqEjNOUhEaJkcXfHRAQwJcS2Ds,27309
-replay/data/nn/sequential_dataset.py,sha256=DqV7inCeDH_RHsx8ZyO1NC9p1V-Q_j0MNJMp_NnV0tM,8641
-replay/data/nn/torch_sequential_dataset.py,sha256=5APjkLVdX-e0--ymu_BSkBvfMPVyuQ_F3F6MdRmLFOI,10859
-replay/data/nn/utils.py,sha256=bE9He2LH_dP5LpRLuWxqMQdGAQCHFynAj-lfJT1NXP4,2801
-replay/data/schema.py,sha256=LiDlcofVtkqXvmtVxoXdoOyNLiv_QOSdRaK1IYJ3FGs,15769
-replay/data/spark_schema.py,sha256=jb1E7D4qupurxuhOhEhfGrSr-pGd3y8zSelvSeyl1g8,1148
-replay/metrics/__init__.py,sha256=KDkxVnKa4ks9K9GmlrdTx1pkIl-MAmm78ZASsp2ZndE,2812
-replay/metrics/base_metric.py,sha256=7nWXeyp7-qD2QYqwF7oEsDiiyH95JtRWxylEsAfWGL4,16615
-replay/metrics/categorical_diversity.py,sha256=pj3l3E_COhbxwJzsz1wxoSbk3tVcDRLEf3B5jDOSmW4,11591
-replay/metrics/coverage.py,sha256=EF0cWUvgCG8h2EbIlUly8iVfOtMJ8gMLB-FcVl86dsM,9349
-replay/metrics/descriptors.py,sha256=UX7GlWLE8SERg-Rf_rURzzWYAUu394OmJrfigbR-sew,4006
-replay/metrics/experiment.py,sha256=YNsa48UVPmXE6nePwSjxwVJMmd5U8cFJo-vJsxlaymw,8309
-replay/metrics/hitrate.py,sha256=9ZExNCcTWcBPKdDDLm35cxTGCHUk-BW1cLkFuyK01a8,2426
-replay/metrics/map.py,sha256=M-vKyifDnSA4HtbxYm6oxZ_ug9mOYdLH-mQahaq0gxU,2684
-replay/metrics/mrr.py,sha256=cC0YIgkmiV5RYF3dv1sd1sc-jVbTC0gswkGFU6egaf8,2279
-replay/metrics/ndcg.py,sha256=Ez01dPjLfyrbLcYdI5lT8R8Ig6haRCzxF_30OOBeu5M,3217
-replay/metrics/novelty.py,sha256=PbjvQkSNs_QfZXs24mDEcpRFrOoN_CDGKr3A-U3w9wU,5815
-replay/metrics/offline_metrics.py,sha256=4K_TZodkVbvKCCU1DGdmskA1L-Q3ECovcwlzXVW_nLc,21295
-replay/metrics/precision.py,sha256=DlK8qy_JdTFEkDFg01yAq9vgCIyCXs3B8ja3KjRH1PI,2369
-replay/metrics/recall.py,sha256=p62JBUNm9SiRi4yUJPkfSqMw2wg9697CUZpb9MifeFc,2560
-replay/metrics/rocauc.py,sha256=2p3wwrrHb7AnXPblebYujjRMVrz5QOS8eRnl5JDn_40,3389
-replay/metrics/surprisal.py,sha256=JAYngAgNPm8OTcjnk05GtGU1VTWc24s-ablktZ6NGJY,7852
-replay/metrics/torch_metrics_builder.py,sha256=FNflawLrFpuTrTqOOz6GJpobH3D0h03riTignTBbqw8,14038
-replay/metrics/unexpectedness.py,sha256=RveOm09Vmo585ELRHX3Sr3b9ZeGgLYFBQxfCuxJ2S7U,7122
-replay/models/__init__.py,sha256=gImNVFNUAZS7NYycDYF9I3H58DO6N7XxAuTye8aI6DQ,840
-replay/models/als.py,sha256=2AqSRvoXJhp_87jKj5zO7KyXsq9neH9Fa4WpFmwaBLg,6401
-replay/models/association_rules.py,sha256=6CE69RyvRLZTFaAS9mqKQxu2v9Fyhlhtw42azxAgmUY,14989
-replay/models/base_neighbour_rec.py,sha256=YPj9eHOXL4KTbp5UBbMXXT8bon-N0wUy02gZNk-V-Is,8077
-replay/models/base_rec.py,sha256=YgQI_vK7K1NMGC7Opca8wSXU65MGwPcIrh5XE5QHjE4,68541
-replay/models/cat_pop_rec.py,sha256=2WvmTnUmPJ-gx8oVZFT3BDtHjOR-B6YyTer2ptbvWV0,12162
-replay/models/cluster.py,sha256=CV847iIFyUDqPSmVrawx7yRwudQoriaA9bvrYS69xmQ,5202
-replay/models/extensions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-replay/models/extensions/ann/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-replay/models/extensions/ann/ann_mixin.py,sha256=y4VahoBipUwZKYCXBXnOOZdCCNTRoXaGWZhge9ZD_uM,7410
-replay/models/extensions/ann/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-replay/models/extensions/ann/entities/base_hnsw_param.py,sha256=AJwM7yEksjP4GWsIDPFtE7JNULp3Uo5J1pGgbifPIrY,808
-replay/models/extensions/ann/entities/hnswlib_param.py,sha256=H80f-BznejRRi-pYtBD7-z1-IsP6GEGX6WAITEdS6uo,2279
-replay/models/extensions/ann/entities/nmslib_hnsw_param.py,sha256=baO8XuUxTNuDJ1JSfoVnaCJW3DSOB7zJFhDaa-IUdLM,2337
-replay/models/extensions/ann/index_builders/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-replay/models/extensions/ann/index_builders/base_index_builder.py,sha256=Ul25G0FaNLOXUjrDXxZDTg7tLXlv1N6wR8kWjWICtZ0,2110
-replay/models/extensions/ann/index_builders/driver_hnswlib_index_builder.py,sha256=Ydzi0Wx9GqiBb065lnSHrsyogqQoLqWeD031k485pMI,1740
-replay/models/extensions/ann/index_builders/driver_nmslib_index_builder.py,sha256=VE_b8elPTBignc6x_jETOzvijXsDT-UjpERQbWf_ays,1359
-replay/models/extensions/ann/index_builders/executor_hnswlib_index_builder.py,sha256=2HP9EQQXA0YrgssRnEqFDfkCn5MgpPbKO8TCJehymlI,2607
-replay/models/extensions/ann/index_builders/executor_nmslib_index_builder.py,sha256=ku-CbdWvGjP_cyF_o2C5G7ZGwcbZtw0fOFn8IckKtB0,2697
-replay/models/extensions/ann/index_builders/nmslib_index_builder_mixin.py,sha256=mWsvnyFoNu1m_oBR-lBFBnBikw76VWOzThOfwCmzVj4,2027
-replay/models/extensions/ann/index_inferers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-replay/models/extensions/ann/index_inferers/base_inferer.py,sha256=GmcEfVBRvhEFgZftms3SszwZ5pSDF-oa6Xv42DZcRc4,2639
-replay/models/extensions/ann/index_inferers/hnswlib_filter_index_inferer.py,sha256=hj-d4bCtQn9S1C4UoSjHSy_BiDDeck9iJJRp724x3Ug,2661
-replay/models/extensions/ann/index_inferers/hnswlib_index_inferer.py,sha256=tGJ3b26j15tWtLtuPsBhcm-cr43r6lzbgR8gWaInVsc,1855
-replay/models/extensions/ann/index_inferers/nmslib_filter_index_inferer.py,sha256=a4F6bBmMpD1A-r-UK3_eq4qFBz1euMhDq2yrWS1YIN8,3422
-replay/models/extensions/ann/index_inferers/nmslib_index_inferer.py,sha256=JTBj5eAX5OP3FM1HRa-wUAasvAKwfD9rC_JH_ZPbUlU,2367
-replay/models/extensions/ann/index_inferers/utils.py,sha256=taAnxnSpvkpFqSUjneZJLRC58Af6Bd0dEnfUZUEwYNU,843
-replay/models/extensions/ann/index_stores/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-replay/models/extensions/ann/index_stores/base_index_store.py,sha256=u4l2ybAXX92ZMGK7NqqosbKF75QgFqhAMVadd5ePj6Y,910
-replay/models/extensions/ann/index_stores/hdfs_index_store.py,sha256=SGXmyy5hl4XgEJJBSpMLyoZBfIlPtZVnsRy07Z5lSM8,3150
-replay/models/extensions/ann/index_stores/shared_disk_index_store.py,sha256=qhwLCJJGoJAH8KoGDU12dW759Qr6Azf0jXCi-b731x8,2066
-replay/models/extensions/ann/index_stores/spark_files_index_store.py,sha256=i4Yz68Vryhalp2z5Jfq8rBRRsSnk8rq6a5w0P5rRO5E,3753
-replay/models/extensions/ann/index_stores/utils.py,sha256=tnVLQEKVcx2xhY6nHACXQF4-xupDO2bVMjHrWAsVywU,3725
-replay/models/extensions/ann/utils.py,sha256=NVekVdx0nHuNP9ehuCFg6ZCj39Gh9Rwy2Nw7tl2eMS0,1350
-replay/models/kl_ucb.py,sha256=lSQUaFLVQS42kR7nhrahGTYPh_xunvgyt0ejDeO4iWM,6849
-replay/models/knn.py,sha256=X9lUlwDzWT5DMt8qI3fheUMEdQLjmt5MgBnJ5ipA_Xo,9576
-replay/models/nn/__init__.py,sha256=AT3o1qXaxUq4_QIGlcGuSs54ZpueOo-SbpZwuGI-6os,41
-replay/models/nn/optimizer_utils/__init__.py,sha256=zafOGP_F7FvwtDbuu8P7fWtM0zMHLwdsQpanw-1tafM,218
-replay/models/nn/optimizer_utils/optimizer_factory.py,sha256=olvfhbQ4bj97reMq6cDC5arKxVgr7lz2vZ6vkj-Gc5o,3028
-replay/models/nn/sequential/__init__.py,sha256=CI2n0cxs_amqJrwBMq6n0Z_uBOu7CGXfagqvE4Jlmjw,128
-replay/models/nn/sequential/bert4rec/__init__.py,sha256=1gFtAOqubZXIaqo4MFv1OVDJtuDzu2L1zS5BAz9O-gM,399
-replay/models/nn/sequential/bert4rec/dataset.py,sha256=jeO-Xndh-Q7tXX_BfLKWphHAdC_ubc7SEe0-RjqKlDA,10410
-replay/models/nn/sequential/bert4rec/lightning.py,sha256=Z0i5GYyjRr9SpHS3LBPBx0isRRdzy0jjBQDcsSspros,22611
-replay/models/nn/sequential/bert4rec/model.py,sha256=K0mmCYR-va7gcXvWwTUKged0Spa9obJXNleDMoEDEkg,21645
-replay/models/nn/sequential/callbacks/__init__.py,sha256=ElQw3etd9iTlnD9YB_Tgh3heAKut_UBC3hXWATkWD20,281
-replay/models/nn/sequential/callbacks/prediction_callbacks.py,sha256=aAfjOqHxn1iqOmeVcpgY8NC8WCmcnrtV0r2xU8F5NKU,9139
-replay/models/nn/sequential/callbacks/validation_callback.py,sha256=YiPalwD1ZE9bWadSC5SqOVjxHBXNTQEmj3dw2oLrEBc,5980
-replay/models/nn/sequential/postprocessors/__init__.py,sha256=89LGzkNHukcuC2-rfpiz7vmv1zyk6MNY-8zaXrvtn0M,164
-replay/models/nn/sequential/postprocessors/_base.py,sha256=Z-HuXM4WBQtrq9IeCL00jtdb_UopksInxVVK8lxD_hA,1136
-replay/models/nn/sequential/postprocessors/postprocessors.py,sha256=GUO5MicViJqbGNCe5nLZafeu5VyrH8Bbp2Ns-ZjSysI,6821
-replay/models/nn/sequential/sasrec/__init__.py,sha256=c6130lRpPkcbuGgkM7slagBIgH7Uk5zUtSzFDEwAsik,250
-replay/models/nn/sequential/sasrec/dataset.py,sha256=VpmoxYPpsj_q97x-Whuebw2ntkGBLTRWWIgfBAYO7Bg,7068
-replay/models/nn/sequential/sasrec/lightning.py,sha256=KGFSoWVJJnCVfvpduafyUn2RqIFhwETGoaYEiZrI0HI,21287
-replay/models/nn/sequential/sasrec/model.py,sha256=rR_gzSeSw27C17T6vY7zkIHfdqk7jBpA1gJVwmp7Lxs,28183
-replay/models/pop_rec.py,sha256=jmxmpz1HjSOEbRTGQYm3ypDUFJCoaAHYLz70KGWkjRM,4972
-replay/models/query_pop_rec.py,sha256=SMdd0ll8JlZMz5waxg0RWEHz2jvyRvV60R7ur3apCCw,4186
-replay/models/random_rec.py,sha256=T48XNRlB3HYbdtYe0DkjWPUN7BA2WOrKzA7RvZbFDWg,8230
-replay/models/slim.py,sha256=WHhSMx5hWHcbU5moCTjZ-MSOUvTR4fzLjUoeIYY7jXE,4863
-replay/models/thompson_sampling.py,sha256=Dq919ydM7Rexq9sTlUTtcFdIUBwHveoKQEyANUg_IdQ,1959
-replay/models/ucb.py,sha256=CYUnrhEyJnukAW94ymBE7_Ylw9uL0y9HvXTHTxASkrw,7081
-replay/models/wilson.py,sha256=vKEuK3M4zPKwPt-nt6PBNtlpC2vjieSR-hgMLySllu4,4711
-replay/models/word2vec.py,sha256=Jnq_yQ1CucebMCZXybOexccl5BnRa4_8fR_QtS08Drg,10170
-replay/optimization/__init__.py,sha256=az6U10rF7X6rPRUUPwLyiM1WFNJ_6kl0imA5xLVWFLs,120
-replay/optimization/optuna_objective.py,sha256=p-ja-FBFxTD81KdgCbWvEw-F3WXNqIJZAdd4re906iM,7990
-replay/preprocessing/__init__.py,sha256=NNceamEWs9J7No08_-7Gj6AaThgDHhlBbw3SRhXYKJU,412
-replay/preprocessing/converter.py,sha256=feCKZ1xk-DPAQfxmuDO0m47Eq0168TRxcaeCf4kgaf4,4513
-replay/preprocessing/filters.py,sha256=yYptV7CCOtZsbplkH_g3rp1yMIMVMyf9RKWE1kWS0Ng,35195
-replay/preprocessing/history_based_fp.py,sha256=ROUxmEmWDGIRJYo_8RSW5zCq45Aaco2retPZtlzowh0,19910
-replay/preprocessing/label_encoder.py,sha256=-A_hSxNuz-ssOB3pcl6dV61OyPQileYorcKKUS31xFQ,25513
-replay/preprocessing/sessionizer.py,sha256=_rTF8Tv9qJ_8_RdOb3kqb6S05klQxqnqKl5BlMoK0mw,12530
-replay/scenarios/__init__.py,sha256=kw2wRkPPinw0IBA20D83XQ3xeSudk3KuYAAA1Wdr8xY,93
-replay/scenarios/fallback.py,sha256=8T3zGQXUsqwAxTE3ed5lKJA1-mZw6OPWD4Me8XsYFMc,7357
-replay/splitters/__init__.py,sha256=DnqVMelrzLwR8fGQgcWN_8FipGs8T4XGSPOMW-L_x2g,454
-replay/splitters/base_splitter.py,sha256=hFFibL7wxTMDyFhwoIAwm7qmtdQM5mxIyc81qP7bL8w,6958
-replay/splitters/cold_user_random_splitter.py,sha256=vHoq3o1fZM0ECqeNG5GPhlShjxYSQj_4DjMduaKdX0I,4406
-replay/splitters/k_folds.py,sha256=PlsMlM7l5Wl_71PJamIb0Si_tG8s13StZ1-1GTuOZg4,6541
-replay/splitters/last_n_splitter.py,sha256=PRWgUo39OCYXSPpMxiXsfBVJXWr4LcUIuYsvUaNDQ2o,15664
-replay/splitters/new_users_splitter.py,sha256=V2IX886N1FFTqnzhGuQHi3g2UELtFdal2Zj68GXpR2A,9620
-replay/splitters/random_splitter.py,sha256=hLOvhBUzrfS5mdKb5-0I3v2XroFO5wHuds24d0WLhQ4,3121
-replay/splitters/ratio_splitter.py,sha256=BsEbha9uPFXKcEQP6qAr_hv6hU3otCpnuUgKoReYDFg,18158
-replay/splitters/time_splitter.py,sha256=ON_kVLcIRVcrQVMhzT0snNRTmHRPvwik7NysLc_mPMg,9251
-replay/splitters/two_stage_splitter.py,sha256=7tde_UsccuwmUAquMt5Ci3cM0sV09JObI3jRPa4Y5WI,18419
-replay/utils/__init__.py,sha256=tubXIraBtiw9AeONaOxv6KMbaQCrDr3ukv4RCdfTIkk,253
-replay/utils/dataframe_bucketizer.py,sha256=am1AtlZApUXU5xp4kZeQNWxQmpfkONDSxTrP-Coz8xo,3834
-replay/utils/distributions.py,sha256=i3DQ4fQlhNs-J5C3neQk2zmPWfLwrSu4ys3InRr1gNk,1304
-replay/utils/model_handler.py,sha256=jF8FTzHAKyjY4j-5sRHDk47fYuTzLNWE6mGnNR6zgFk,5859
-replay/utils/session_handler.py,sha256=IUhm7Da3cgF0qg0rlEo9NiDnpGBB267cWzEGqwzRmmY,5359
-replay/utils/spark_utils.py,sha256=4db-XMvKnnPGOzVngIdTqOBX0-1zPDLwL0MrC1lUaVY,27061
-replay/utils/time.py,sha256=pEOqOXzAlj2F8nqK4yCG9OakwMzQqQl6FuUTkdp6svk,9467
-replay/utils/types.py,sha256=vdQWLwY9X1lxSor9kaw8U1jIDcBkUAGbwfIzEBoVkrU,744
-replay_rec-0.16.0.dist-info/LICENSE,sha256=rPmcA7UrHxBChEAAlJyE24qUWKKl9yLQXxFsKeg_LX4,11344
-replay_rec-0.16.0.dist-info/METADATA,sha256=kqBuaPryc61S_1BllumtPygYrGjkS4BvEmS0A_HAFts,10597
-replay_rec-0.16.0.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-replay_rec-0.16.0.dist-info/RECORD,,