keras-rs-nightly 0.0.1.dev2025021903__py3-none-any.whl → 0.3.1.dev202512130338__py3-none-any.whl

keras_rs/__init__.py

@@ -1,30 +1,11 @@
-import os
+"""DO NOT EDIT.
 
-# Import everything from /api/ into keras_rs.
-from keras_rs.api import *  # noqa: F403
+This file was autogenerated. Do not edit it by hand,
+since your modifications would be overwritten.
+"""
 
-# Import * ignores names starting with "_", and `__version__` comes from
-# `version` anyway.
-from keras_rs.src.version import __version__
-
-# Add everything in /api/ to the module search path.
-__path__.append(os.path.join(os.path.dirname(__file__), "api"))  # noqa: F405
-
-# Don't pollute namespace.
-del os
-
-
-# Never autocomplete `.src` or `.api` on an imported keras_rs object.
-def __dir__() -> list[str]:
-    keys = dict.fromkeys((globals().keys()))
-    keys.pop("src")
-    keys.pop("api")
-    return list(keys)
-
-
-# Don't import `.src` or `.api` during `from keras_rs import *`.
-__all__ = [
-    name
-    for name in globals().keys()
-    if not (name.startswith("_") or name in ("src", "api"))
-]
+from keras_rs import layers as layers
+from keras_rs import losses as losses
+from keras_rs import metrics as metrics
+from keras_rs.src.version import __version__ as __version__
+from keras_rs.src.version import version as version

keras_rs/layers/__init__.py

@@ -0,0 +1,37 @@
+"""DO NOT EDIT.
+
+This file was autogenerated. Do not edit it by hand,
+since your modifications would be overwritten.
+"""
+
+from keras_rs.src.layers.embedding.distributed_embedding import (
+    DistributedEmbedding as DistributedEmbedding,
+)
+from keras_rs.src.layers.embedding.distributed_embedding_config import (
+    FeatureConfig as FeatureConfig,
+)
+from keras_rs.src.layers.embedding.distributed_embedding_config import (
+    TableConfig as TableConfig,
+)
+from keras_rs.src.layers.embedding.embed_reduce import (
+    EmbedReduce as EmbedReduce,
+)
+from keras_rs.src.layers.feature_interaction.dot_interaction import (
+    DotInteraction as DotInteraction,
+)
+from keras_rs.src.layers.feature_interaction.feature_cross import (
+    FeatureCross as FeatureCross,
+)
+from keras_rs.src.layers.retrieval.brute_force_retrieval import (
+    BruteForceRetrieval as BruteForceRetrieval,
+)
+from keras_rs.src.layers.retrieval.hard_negative_mining import (
+    HardNegativeMining as HardNegativeMining,
+)
+from keras_rs.src.layers.retrieval.remove_accidental_hits import (
+    RemoveAccidentalHits as RemoveAccidentalHits,
+)
+from keras_rs.src.layers.retrieval.retrieval import Retrieval as Retrieval
+from keras_rs.src.layers.retrieval.sampling_probability_correction import (
+    SamplingProbabilityCorrection as SamplingProbabilityCorrection,
+)

keras_rs/losses/__init__.py

@@ -0,0 +1,19 @@
+"""DO NOT EDIT.
+
+This file was autogenerated. Do not edit it by hand,
+since your modifications would be overwritten.
+"""
+
+from keras_rs.src.losses.list_mle_loss import ListMLELoss as ListMLELoss
+from keras_rs.src.losses.pairwise_hinge_loss import (
+    PairwiseHingeLoss as PairwiseHingeLoss,
+)
+from keras_rs.src.losses.pairwise_logistic_loss import (
+    PairwiseLogisticLoss as PairwiseLogisticLoss,
+)
+from keras_rs.src.losses.pairwise_mean_squared_error import (
+    PairwiseMeanSquaredError as PairwiseMeanSquaredError,
+)
+from keras_rs.src.losses.pairwise_soft_zero_one_loss import (
+    PairwiseSoftZeroOneLoss as PairwiseSoftZeroOneLoss,
+)

keras_rs/metrics/__init__.py

@@ -0,0 +1,16 @@
+"""DO NOT EDIT.
+
+This file was autogenerated. Do not edit it by hand,
+since your modifications would be overwritten.
+"""
+
+from keras_rs.src.metrics.dcg import DCG as DCG
+from keras_rs.src.metrics.mean_average_precision import (
+    MeanAveragePrecision as MeanAveragePrecision,
+)
+from keras_rs.src.metrics.mean_reciprocal_rank import (
+    MeanReciprocalRank as MeanReciprocalRank,
+)
+from keras_rs.src.metrics.ndcg import NDCG as NDCG
+from keras_rs.src.metrics.precision_at_k import PrecisionAtK as PrecisionAtK
+from keras_rs.src.metrics.recall_at_k import RecallAtK as RecallAtK