🐦
pigeon - Quickly annotate data on Jupyter
Pigeon is a simple widget that lets you quickly annotate a dataset of unlabeled examples from the comfort of your Jupyter notebook.
Pigeon currently supports annotation for classification tasks (set of labels), regression tasks (int/float range), or captioning tasks (variable-length text). Anything that can be displayed on Jupyter (text, images, audio, graphs, etc.) can be displayed by pigeon by providing the appropriate display_fn
argument.
Installation
pip install pigeon-jupyter
Examples
- Text classification
Code:
from pigeon import annotate
annotations = annotate(
['I love this movie', 'I was really disappointed by the book'],
options=['positive', 'negative']
)
Preview:
- Image classification
Code:
from pigeon import annotate
from IPython.display import display, Image
annotations = annotate(
['assets/img_example1.jpg', 'assets/img_example2.jpg'],
options=['cat', 'dog', 'horse'],
display_fn=lambda filename: display(Image(filename))
)
Preview: