fit_predict_proba
crowdkit.aggregation.classification.gold_majority_vote.GoldMajorityVote.fit_predict_proba
| Source code
fit_predict_proba(
self,
data: DataFrame,
true_labels: Series
)
Fit the model and return probability distributions on labels for each task.
Parameters Description
Parameters | Type | Description |
---|---|---|
data |
DataFrame | Workers' labeling results. A pandas.DataFrame containing |
true_labels |
Series | Tasks' ground truth labels. A pandas.Series indexed by |
-
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