Rax
stable
🦖 Rax Documentation
API Documentation
Ranking Losses (
rax.*_loss
)
Ranking Metrics (
rax.*_metric
)
Function Transformations (
rax.*_t12n
)
Lambdaweights (
rax.*_lambdaweight
)
Utilities
Types
References
Examples
Approximate Metric Optimization
Flax Integration
Segmentation
T5X Integration
Rax
Index
Edit on GitHub
Index
_
|
A
|
B
|
C
|
D
|
G
|
L
|
M
|
N
|
P
|
R
|
S
|
U
_
__call__() (rax.types.CutoffFn method)
(rax.types.LambdaweightFn method)
(rax.types.LossFn method)
(rax.types.MetricFn method)
(rax.types.RankFn method)
(rax.types.ReduceFn method)
A
ap_metric() (in module rax)
approx_cutoff() (in module rax.utils)
approx_ranks() (in module rax.utils)
approx_t12n() (in module rax)
B
bound_t12n() (in module rax)
C
cutoff() (in module rax.utils)
CutoffFn (class in rax.types)
D
dcg2_lambdaweight() (in module rax)
dcg_lambdaweight() (in module rax)
dcg_metric() (in module rax)
G
gumbel_t12n() (in module rax)
L
labeldiff_lambdaweight() (in module rax)
LambdaweightFn (class in rax.types)
listmle_loss() (in module rax)
LossFn (class in rax.types)
M
MetricFn (class in rax.types)
module
rax._src.lambdaweights
rax._src.losses
rax._src.metrics
rax._src.t12n
rax._src.types
mrr_metric() (in module rax)
N
ndcg_metric() (in module rax)
P
pairwise_hinge_loss() (in module rax)
pairwise_logistic_loss() (in module rax)
pairwise_mse_loss() (in module rax)
pairwise_qr_loss() (in module rax)
pairwise_soft_zero_one_loss() (in module rax)
pointwise_mse_loss() (in module rax)
pointwise_sigmoid_loss() (in module rax)
poly1_softmax_loss() (in module rax)
precision_metric() (in module rax)
R
RankFn (class in rax.types)
ranks() (in module rax.utils)
rax._src.lambdaweights
module
rax._src.losses
module
rax._src.metrics
module
rax._src.t12n
module
rax._src.types
module
recall_metric() (in module rax)
ReduceFn (class in rax.types)
S
segment_t12n() (in module rax)
softmax_loss() (in module rax)
U
unique_softmax_loss() (in module rax)
Read the Docs
v: stable
Versions
latest
stable
Downloads
On Read the Docs
Project Home
Builds