fit_predict
crowdkit.aggregation.embeddings.closest_to_average.ClosestToAverage.fit_predict
| Source code
fit_predict(
self,
data: DataFrame,
aggregated_embeddings: Optional[Series] = None
)
Fit the model and return the aggregated results.
Parameters Description
Parameters | Type | Description |
---|---|---|
data |
DataFrame | Workers' outputs with their embeddings. A pandas.DataFrame containing |
aggregated_embeddings |
Optional[Series] | Tasks' embeddings. A pandas.Series indexed by |
-
Returns:
Tasks' embeddings and outputs. A pandas.DataFrame indexed by
task
withembedding
andoutput
columns. -
Return type:
DataFrame