fit

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

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

Fits the model.

Parameters Description

Parameters Type Description
data DataFrame

Workers' outputs with their embeddings. A pandas.DataFrame containing task, worker, output and embedding columns.

aggregated_embeddings Optional[Series]

Tasks' embeddings. A pandas.Series indexed by task and holding corresponding embeddings.

true_embeddings Optional[Series]

Tasks' embeddings. A pandas.Series indexed by task and holding corresponding embeddings.