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