keras-rs-nightly 0.0.1.dev2025030403__py3-none-any.whl → 0.0.1.dev2025030503__py3-none-any.whl

keras_rs/api/layers/__init__.py

@@ -12,3 +12,6 @@ from keras_rs.src.layers.retrieval.brute_force_retrieval import (
 from keras_rs.src.layers.retrieval.hard_negative_mining import (
     HardNegativeMining,
 )
+from keras_rs.src.layers.retrieval.sampling_probability_correction import (
+    SamplingProbabilityCorrection,
+)

keras_rs/src/layers/retrieval/sampling_probability_correction.py

@@ -0,0 +1,48 @@
+from typing import Any
+
+import keras
+from keras import ops
+
+from keras_rs.src import types
+from keras_rs.src.api_export import keras_rs_export
+
+
+@keras_rs_export("keras_rs.layers.SamplingProbabilityCorrection")
+class SamplingProbabilityCorrection(keras.layers.Layer):
+    """Sampling probability correction.
+
+    Corrects the logits to reflect the sampling probability of negatives.
+
+    Args:
+        epsilon: float. Small float added to sampling probability to avoid
+            taking the log of zero. Defaults to 1e-6.
+        **kwargs: Args to pass to the base class.
+    """
+
+    def __init__(self, epsilon: float = 1e-6, **kwargs: Any) -> None:
+        super().__init__(**kwargs)
+        self.epsilon = epsilon
+        self.built = True
+
+    def call(
+        self,
+        logits: types.Tensor,
+        candidate_sampling_probability: types.Tensor,
+    ) -> types.Tensor:
+        """Corrects input logits to account for candidate sampling probability.
+
+        Args:
+            logits: The logits to correct.
+            candidate_sampling_probability: The sampling probability.
+
+        Returns:
+            The corrected logits.
+        """
+        return logits - ops.log(
+            ops.clip(candidate_sampling_probability, self.epsilon, 1.0)
+        )
+
+    def get_config(self) -> dict[str, Any]:
+        config: dict[str, Any] = super().get_config()
+        config.update({"epsilon": self.epsilon})
+        return config

keras_rs/src/version.py

@@ -1,7 +1,7 @@
 from keras_rs.src.api_export import keras_rs_export
 
 # Unique source of truth for the version number.
-__version__ = "0.0.1.dev2025030403"
+__version__ = "0.0.1.dev2025030503"
 
 
 @keras_rs_export("keras_rs.version")

{keras_rs_nightly-0.0.1.dev2025030403.dist-info → keras_rs_nightly-0.0.1.dev2025030503.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: keras-rs-nightly
-Version: 0.0.1.dev2025030403
+Version: 0.0.1.dev2025030503
 Summary: Multi-backend recommender systems with Keras 3.
 Author-email: Keras RS team <keras-rs@google.com>
 License: Apache License 2.0

{keras_rs_nightly-0.0.1.dev2025030403.dist-info → keras_rs_nightly-0.0.1.dev2025030503.dist-info}/RECORD

@@ -1,10 +1,10 @@
 keras_rs/__init__.py,sha256=X3VNKb_6VDEs5GqcbEc_l8mAsefWb5UgSu8krnQdFcM,794
 keras_rs/api/__init__.py,sha256=BU34SgAwrjZh49ppXGxJPxreTOYW0C7vy3_x8nmvHUk,240
-keras_rs/api/layers/__init__.py,sha256=ucB0_mJ9qnZWQp_RLipoY20YRMskNUB9ril9ZxZ3lkM,446
+keras_rs/api/layers/__init__.py,sha256=MOWXxHFJQKriffbD-k8giN87CXdzrbRwayzmDipoBIo,559
 keras_rs/src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_rs/src/api_export.py,sha256=RsmG-DvO-cdFeAF9W6LRzms0kvtm-Yp9BAA_d-952zI,510
 keras_rs/src/types.py,sha256=UyOdgjqrqg_b58opnY8n6gTiDHKVR8z_bmEruehERBk,514
-keras_rs/src/version.py,sha256=1GdZ3-GmwS0fWXnm8IFMz6QbfvrlMK4CVsWJXoftvFk,222
+keras_rs/src/version.py,sha256=aGxhmch--1ygcaY54yqn6fBeP77KQEG3cXmhlwv_c-4,222
 keras_rs/src/layers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_rs/src/layers/modeling/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_rs/src/layers/modeling/dot_interaction.py,sha256=jGHcg0EiWxth6LTxG2yWgHcyx_GXrxvA61uQqpPfnDQ,6900
@@ -12,9 +12,10 @@ keras_rs/src/layers/modeling/feature_cross.py,sha256=5OCSI0vFYzJNmgkKcuHIbVv8U2q
 keras_rs/src/layers/retrieval/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_rs/src/layers/retrieval/brute_force_retrieval.py,sha256=mohILOt6PC6jHBztaowDbj3QBnSetuvkq55FmE39PlY,7321
 keras_rs/src/layers/retrieval/hard_negative_mining.py,sha256=8c44iEUmK_SwfHPdrVtg96ycBZzYf62ee66a49mqCbU,4115
+keras_rs/src/layers/retrieval/sampling_probability_correction.py,sha256=80vgOPfBiF-PC0dSyqS57IcIxOxi_Q_R7eSXHn1G0yI,1437
 keras_rs/src/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_rs/src/utils/keras_utils.py,sha256=IjWSRieBkv7UX12qgUoI1tcOeISstCLRSTqSHpT06yE,1275
-keras_rs_nightly-0.0.1.dev2025030403.dist-info/METADATA,sha256=kJQ7L75QaEg2s8chc67UW1aNZfZiokOdFRgRMhSFYC8,3522
-keras_rs_nightly-0.0.1.dev2025030403.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-keras_rs_nightly-0.0.1.dev2025030403.dist-info/top_level.txt,sha256=pWs8X78Z0cn6lfcIb9VYOW5UeJ-TpoaO9dByzo7_FFo,9
-keras_rs_nightly-0.0.1.dev2025030403.dist-info/RECORD,,
+keras_rs_nightly-0.0.1.dev2025030503.dist-info/METADATA,sha256=AA3mZmkM2S2joC0BI5EcuM_kwFk61MPMjqYEXYoguuE,3522
+keras_rs_nightly-0.0.1.dev2025030503.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
+keras_rs_nightly-0.0.1.dev2025030503.dist-info/top_level.txt,sha256=pWs8X78Z0cn6lfcIb9VYOW5UeJ-TpoaO9dByzo7_FFo,9
+keras_rs_nightly-0.0.1.dev2025030503.dist-info/RECORD,,