fit_predict

crowdkit.aggregation.image_segmentation.segmentation_majority_vote.SegmentationMajorityVote.fit_predict | Source code

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

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'.

skills Optional[Series]

workers' skills. A pandas.Series index by workers and holding corresponding worker's skill

  • Returns:

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

  • Return type:

    Series