snorkel.utils.preds_to_probs

snorkel.utils.preds_to_probs(preds, num_classes)

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