predict_proba

crowdkit.aggregation.classification.zero_based_skill.ZeroBasedSkill.predict_proba | Source code

predict_proba(self, data: DataFrame)

Returns probability distributions of labels for each task when the model is fitted.

Parameters description

ParametersTypeDescription
dataDataFrame

The training dataset of workers' labeling results which is represented as the pandas.DataFrame data containing task, worker, and label columns.

  • Returns:

    The probability distributions of task labels. The pandas.DataFrame data is indexed by task so that result.loc[task, label] is the probability that the task true label is equal to label. Each probability is in the range from 0 to 1, all task probabilities must sum up to 1.

  • Return type:

    DataFrame

Last updated: March 31, 2023

Crowd-Kit
Overview
Reference
Aggregation
Datasets
Learning
Metrics
Postprocessing