Toloka documentation

fit_predict

crowdkit.aggregation.image_segmentation.segmentation_em.SegmentationEM.fit_predict | Source code

fit_predict(self, data: DataFrame)

Fit the model and return the aggregated segmentations.

Parameters Description

Parameters Type Description
data DataFrame

Workers' segmentations. A pandas.DataFrame containing worker, task and segmentation columns'.

  • Returns:

    Tasks' segmentations. A pandas.Series indexed by task such that labels.loc[task] is the tasks's aggregated segmentation.

  • Return type:

    Series