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.
pip install pigeon-jupyter- 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:
