fit_predict

crowdkit.aggregation.embeddings.closest_to_average.ClosestToAverage.fit_predict | Source code

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

Fits the model to the training data and returns the aggregated outputs.

Parameters description

Parameters	Type	Description
`data`	DataFrame	The workers' outputs with their embeddings. The `pandas.DataFrame` data contains the `task`, `worker`, `output`, and `embedding` columns.
`aggregated_embeddings`	Optional[Series]	The task aggregated embeddings. The `pandas.Series` data is indexed by `task` and contains the corresponding aggregated embeddings.

Returns:
The task embeddings and outputs. The pandas.DataFrame data is indexed by task and has the embedding and output columns.
Return type:
DataFrame

Crowd-Kit

Reference

Aggregation

Datasets

Learning

Metrics

Postprocessing