Computational quality control for crowdsourcing
Crowd-Kit is a powerful Python library that implements commonly-used aggregation methods for crowdsourced annotation and offers the relevant metrics and datasets. We strive to implement functionality that simplifies working with crowdsourced data.
Currently, Crowd-Kit contains:
To install Crowd-Kit, run the following command:
pip install crowd-kit
If you also want to use the learning
subpackage, type
pip instal crowd-kit[learning]
If you are interested in contributing to Crowd-Kit, use Pipenv to install the library with its dependencies:
pipenv install --dev
We use pytest for testing.
This example shows how to use Crowd-Kit for categorical aggregation using the classical Dawid-Skene algorithm.
First, let us do all the necessary imports.
from crowdkit.aggregation import DawidSkenefrom crowdkit.datasets import load_datasetimport pandas as pd
Then, you need to read your annotations into Pandas DataFrame with columns task
, worker
, label
. Alternatively, you can download an example dataset:
df = pd.read_csv('results.csv') # should contain columns: task, worker, label# df, ground_truth = load_dataset('relevance-2') # or download an example dataset
Then, you can aggregate the workers' responses using the fit_predict
method from the scikit-learn library:
aggregated_labels = DawidSkene(n_iter=100).fit_predict(df)
Below is the list of currently implemented methods, including the already available (✅) and in progress (🟡).
Method | Status |
---|---|
Majority Vote | ✅ |
One-coin Dawid-Skene | ✅ |
Dawid-Skene | ✅ |
Gold Majority Vote | ✅ |
M-MSR | ✅ |
Wawa | ✅ |
Zero-Based Skill | ✅ |
GLAD | ✅ |
KOS | ✅ |
MACE | ✅ |
BCC | 🟡 |
Method | Status |
---|---|
Binary Relevance | ✅ |
Method | Status |
---|---|
Segmentation MV | ✅ |
Segmentation RASA | ✅ |
Segmentation EM | ✅ |
Method | Status |
---|---|
Bradley-Terry | ✅ |
Noisy Bradley-Terry | ✅ |
Method | Status |
---|---|
CrowdLayer | ✅ |
CoNAL | ✅ |
TextSummarization | ✅ |
Stay informed about updates to the platform and open-source libraries — connect with the Toloka team in our Global Community for announcements, discussions, and invites to events.
Last updated: February 6, 2023