torch_concepts.nn.functional.confidence_selection

confidence_selection(c_confidence: Tensor, theta: Tensor) → Tensor

Selects concepts with confidence above a selected threshold.

Parameters:

• c_confidence – Concept confidence scores.

• theta – Threshold to select confident predictions.

Returns:

mask selecting confident predictions.

Return type:

Tensor