snorkel.utils.preds_to_probs¶
-
snorkel.utils.
preds_to_probs
(preds, num_classes)[source]¶ Convert an array of predictions into an array of probabilistic labels.
- Parameters
pred – A [num_datapoints] or [num_datapoints, 1] array of predictions
- Returns
A [num_datapoints, num_classes] array of probabilistic labels with probability of 1.0 in the column corresponding to the prediction
- Return type
np.ndarray