kd.metrics.BinaryAccuracy

kd.metrics.BinaryAccuracy#

class kauldron.metrics.BinaryAccuracy(
*,
logits: typing.Annotated[typing.Any,
<object object at 0x7824c478ba80>] = '__KEY_REQUIRED__',
labels: typing.Annotated[typing.Any,
<object object at 0x7824c478ba80>] = '__KEY_REQUIRED__',
)[source]

Bases: kauldron.metrics.base.Metric

Classification Accuracy for Binary classification tasks.

logits: Annotated[Any, <object object at 0x7824c478ba80>] = '__KEY_REQUIRED__'
labels: Annotated[Any, <object object at 0x7824c478ba80>] = '__KEY_REQUIRED__'
class State(
total: "Float['']",
count: "Float['']",
*,
parent: '_MetricT' = <_EMPTY_TYPE.EMPTY: 1>,
)[source]

Bases: kauldron.metrics.base_state.AverageState

merge(
other: kauldron.metrics.base_state.AverageState,
) kauldron.metrics.base_state.AverageState[source]

Returns a new state that is the accumulation of self and other.

Parameters:

other – A State whose intermediate values should be accumulated onto the values of self.

Returns:

A new State that accumulates the value from both self and other.

replace(**updates)

Returns a new object replacing the specified fields with new values.

get_state(
logits: jaxtyping.Float[Array, '*any'] | jaxtyping.Float[ndarray, '*any'],
labels: jaxtyping.Int[Array, '*any'] | jaxtyping.Int[ndarray, '*any'],
mask: jaxtyping.Bool[Array, '*#any'] | jaxtyping.Bool[ndarray, '*#any'] | jaxtyping.Float[Array, '*#any'] | jaxtyping.Float[ndarray, '*#any'] | None = None,
)[source]
empty() kauldron.metrics.base.Metric.State[source]