predict_proba
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 Description
Parameters | Type | Description |
---|---|---|
data |
DataFrame | Workers' labeling results. A pandas.DataFrame containing |
-
Returns:
Tasks' label probability distributions. A pandas.DataFrame indexed by
task
such thatresult.loc[task, label]
is the probability oftask
's true label to be equal tolabel
. Each probability is between 0 and 1, all task's probabilities should sum up to 1 -
Return type:
DataFrame