crowdkit.aggregation.classification.wawa.Wawa.predict_proba | Source code
predict_proba(self, data: DataFrame)Return probability distributions on labels for each task when the model is fitted.
| Parameters | Type | Description |
|---|---|---|
data | DataFrame | Workers' labeling results. A pandas.DataFrame containing |
Returns:
Tasks' label probability distributions.
A pandas.DataFrame indexed by task such that result.loc[task, label]
is the probability of task's true label to be equal to label. Each
probability is between 0 and 1, all task's probabilities should sum up to 1
Return type:
DataFrame