Skip to content

Commit c0be0bb

Browse files
LakshmiKalaKadaliLakshmiKalaKadali
committed
Regenerate API to include ListMLELoss
1 parent eda8fbb commit c0be0bb

File tree

5 files changed

+23
-50
lines changed

5 files changed

+23
-50
lines changed

‎api_gen.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
import os
1010
import shutil
11-
11+
import pre_commit
1212
import namex
1313

1414
PACKAGE = "keras_rs"

‎keras_rs/api/__init__.py‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,9 @@
44
since your modifications would be overwritten.
55
"""
66

7+
78
from keras_rs import layers as layers
89
from keras_rs import losses as losses
910
from keras_rs import metrics as metrics
10-
from keras_rs.src.version import __version__ as __version__
1111
from keras_rs.src.version import version as version
12+
from keras_rs.src.version import __version__ as __version__

‎keras_rs/api/layers/__init__.py‎

Lines changed: 11 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -4,34 +4,15 @@
44
since your modifications would be overwritten.
55
"""
66

7-
from keras_rs.src.layers.embedding.distributed_embedding import (
8-
DistributedEmbedding as DistributedEmbedding,
9-
)
10-
from keras_rs.src.layers.embedding.distributed_embedding_config import (
11-
FeatureConfig as FeatureConfig,
12-
)
13-
from keras_rs.src.layers.embedding.distributed_embedding_config import (
14-
TableConfig as TableConfig,
15-
)
16-
from keras_rs.src.layers.embedding.embed_reduce import (
17-
EmbedReduce as EmbedReduce,
18-
)
19-
from keras_rs.src.layers.feature_interaction.dot_interaction import (
20-
DotInteraction as DotInteraction,
21-
)
22-
from keras_rs.src.layers.feature_interaction.feature_cross import (
23-
FeatureCross as FeatureCross,
24-
)
25-
from keras_rs.src.layers.retrieval.brute_force_retrieval import (
26-
BruteForceRetrieval as BruteForceRetrieval,
27-
)
28-
from keras_rs.src.layers.retrieval.hard_negative_mining import (
29-
HardNegativeMining as HardNegativeMining,
30-
)
31-
from keras_rs.src.layers.retrieval.remove_accidental_hits import (
32-
RemoveAccidentalHits as RemoveAccidentalHits,
33-
)
7+
8+
from keras_rs.src.layers.embedding.distributed_embedding import DistributedEmbedding as DistributedEmbedding
9+
from keras_rs.src.layers.embedding.distributed_embedding_config import FeatureConfig as FeatureConfig
10+
from keras_rs.src.layers.embedding.distributed_embedding_config import TableConfig as TableConfig
11+
from keras_rs.src.layers.embedding.embed_reduce import EmbedReduce as EmbedReduce
12+
from keras_rs.src.layers.feature_interaction.dot_interaction import DotInteraction as DotInteraction
13+
from keras_rs.src.layers.feature_interaction.feature_cross import FeatureCross as FeatureCross
14+
from keras_rs.src.layers.retrieval.brute_force_retrieval import BruteForceRetrieval as BruteForceRetrieval
15+
from keras_rs.src.layers.retrieval.hard_negative_mining import HardNegativeMining as HardNegativeMining
16+
from keras_rs.src.layers.retrieval.remove_accidental_hits import RemoveAccidentalHits as RemoveAccidentalHits
3417
from keras_rs.src.layers.retrieval.retrieval import Retrieval as Retrieval
35-
from keras_rs.src.layers.retrieval.sampling_probability_correction import (
36-
SamplingProbabilityCorrection as SamplingProbabilityCorrection,
37-
)
18+
from keras_rs.src.layers.retrieval.sampling_probability_correction import SamplingProbabilityCorrection as SamplingProbabilityCorrection

‎keras_rs/api/losses/__init__.py‎

Lines changed: 6 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,9 @@
44
since your modifications would be overwritten.
55
"""
66

7-
from keras_rs.src.losses.pairwise_hinge_loss import (
8-
PairwiseHingeLoss as PairwiseHingeLoss,
9-
)
10-
from keras_rs.src.losses.pairwise_logistic_loss import (
11-
PairwiseLogisticLoss as PairwiseLogisticLoss,
12-
)
13-
from keras_rs.src.losses.pairwise_mean_squared_error import (
14-
PairwiseMeanSquaredError as PairwiseMeanSquaredError,
15-
)
16-
from keras_rs.src.losses.pairwise_soft_zero_one_loss import (
17-
PairwiseSoftZeroOneLoss as PairwiseSoftZeroOneLoss,
18-
)
7+
8+
from keras_rs.src.losses.list_mle_loss import ListMLELoss as ListMLELoss
9+
from keras_rs.src.losses.pairwise_hinge_loss import PairwiseHingeLoss as PairwiseHingeLoss
10+
from keras_rs.src.losses.pairwise_logistic_loss import PairwiseLogisticLoss as PairwiseLogisticLoss
11+
from keras_rs.src.losses.pairwise_mean_squared_error import PairwiseMeanSquaredError as PairwiseMeanSquaredError
12+
from keras_rs.src.losses.pairwise_soft_zero_one_loss import PairwiseSoftZeroOneLoss as PairwiseSoftZeroOneLoss

‎keras_rs/api/metrics/__init__.py‎

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -4,13 +4,10 @@
44
since your modifications would be overwritten.
55
"""
66

7+
78
from keras_rs.src.metrics.dcg import DCG as DCG
8-
from keras_rs.src.metrics.mean_average_precision import (
9-
MeanAveragePrecision as MeanAveragePrecision,
10-
)
11-
from keras_rs.src.metrics.mean_reciprocal_rank import (
12-
MeanReciprocalRank as MeanReciprocalRank,
13-
)
9+
from keras_rs.src.metrics.mean_average_precision import MeanAveragePrecision as MeanAveragePrecision
10+
from keras_rs.src.metrics.mean_reciprocal_rank import MeanReciprocalRank as MeanReciprocalRank
1411
from keras_rs.src.metrics.ndcg import NDCG as NDCG
1512
from keras_rs.src.metrics.precision_at_k import PrecisionAtK as PrecisionAtK
1613
from keras_rs.src.metrics.recall_at_k import RecallAtK as RecallAtK

0 commit comments

Comments
 (0)