fit_predict

crowdkit.aggregation.embeddings.rasa.RASA.fit_predict | Source code

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

Fit the model and return aggregated outputs.

Parameters Description

ParametersTypeDescription
dataDataFrame

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

true_embeddingsOptional[Series]

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

  • Returns:

    Tasks' embeddings and outputs. A pandas.DataFrame indexed by task with embedding and output columns.

  • Return type:

    DataFrame

Crowd-Kit
Overview
Reference
Aggregation
Datasets
Learning
Metrics
Postprocessing