Toloka documentation

fit_predict

crowdkit.aggregation.texts.text_hrrasa.TextHRRASA.fit_predict | Source code

fit_predict(
    self,
    data: DataFrame,
    true_objects: Optional[Series] = None
)

Fit the model and return aggregated texts.

Parameters Description

Parameters Type Description
data DataFrame

Workers' outputs. A pandas.DataFrame containing task, worker and output columns.

true_objects Optional[Series]

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

  • Returns:

    Tasks' texts. A pandas.Series indexed by task such that result.loc[task, text] is the task's text.

  • Return type:

    Series