fit_predict

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

fit_predict(
    self,
    data: DataFrame,
    aggregated_embeddings: Optional[Series] = None
)
Copy text

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

Parameters description

• 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