ott.tools.conformal.OTCP.predict

Contents

ott.tools.conformal.OTCP.predict#

OTCP.predict(x, y_candidates=None, alpha=0.1)[source]#

Conformalize the model’s prediction.

Parameters:
  • x (Array) – Features of shape [..., dim_x].

  • y_candidates (Optional[Array]) – Candidate targets of shape [m, dim_y].

  • alpha (float) – Miscoverage level.

Return type:

Array

Returns:

If y_candidates = None, return an array of shape [..., n_target, dim_y], else a boolean array of shape [..., m].