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

• 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