ott.tools.conformal.OTCP

class ott.tools.conformal.OTCP(model, nonconformity_fn=<built-in function sub>, sinkhorn_output=None, offset=0.0, scale=1.0, calibration_scores=None)

Optimal transport conformal prediction [Klein et al., 2025].

Parameters:

• model (Callable[[Array], Array]) – Fitted model.

• nonconformity_fn (Callable[[Array, Array], Array]) – Multivariate nonconformity score function with a signature (target, prediction) -> score.

• sinkhorn_out – Sinkhorn output computed in fit_transport().

• offset (Array) – Offset used when re-scaling the data.

• scale (Array) – Scale when re-scaling the data.

• calibration_scores (Optional[Array]) – Nonconformity calibration scores computed in calibrate().

• sinkhorn_output (Optional[SinkhornOutput])

Methods

calibrate(x, y)	Compute calibration scores.
fit_transport(x, y[, epsilon, n_target, ...])	Fit the transport map.
get_scores(x, y)	Compute nonconformity scores.
nonconformity_fn(b, /)	Same as a - b.
predict(x[, y_candidates, alpha])	Conformalize the model's prediction.

Attributes

calibration_scores
offset
scale
sinkhorn_output
target_measure	Target measure of shape [n_target, dim_y].
model