crowdkit.aggregation.embeddings.rasa.RASA.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 | Type | Description |
---|---|---|
data | DataFrame | The workers' outputs with their embeddings. The |
true_embeddings | Optional[Series] | The embeddings of the true task responses. The |
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