Toloka documentation

predict

crowdkit.aggregation.classification.gold_majority_vote.GoldMajorityVote.predict | Source code

predict(self, data: DataFrame)

Infer the true labels when the model is fitted.

Parameters Description

Parameters Type Description
data DataFrame

Workers' labeling results. A pandas.DataFrame containing task, worker and label columns.

  • Returns:

    Tasks' labels. A pandas.Series indexed by task such that labels.loc[task] is the tasks's most likely true label.

  • Return type:

    Series