fit_predict_scores

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

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

Fits the model to the training data and returns the estimated scores.

Parameters description

ParametersTypeDescription
dataDataFrame

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

aggregated_embeddingsOptional[Series]

The task aggregated embeddings. The pandas.Series data is indexed by task and contains the corresponding aggregated embeddings.

  • Returns:

    The task label scores. The pandas.DataFrame data is indexed by task so that result.loc[task, label] is a score of label for task.

  • Return type:

    DataFrame

Crowd-Kit
Overview
Reference
Aggregation
Datasets
Learning
Metrics
Postprocessing