fit_predict

crowdkit.aggregation.embeddings.hrrasa.HRRASA.fit_predict | Source code

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

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

Parameters description

ParametersTypeDescription
dataDataFrame

The workers' outputs with their embeddings. The pandas.DataFrame data contains the task, worker, output, and embedding columns.

true_embeddingsOptional[Series]

The embeddings of the true task responses. The pandas.Series data is indexed by task and contains the corresponding embeddings. The multiple true embeddings are not supported for a single task.

  • Returns:

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

  • Return type:

    DataFrame

Last updated: March 31, 2023

Crowd-Kit
Overview
Reference
Aggregation
Datasets
Learning
Metrics
Postprocessing