We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Machine Learning

Title: Small-text: Active Learning for Text Classification in Python

Abstract: We present small-text, a simple modular active learning library, which offers pool-based active learning for text classification in Python. It comes with various pre-implemented state-of-the-art query strategies, including some which can leverage the GPU. Clearly defined interfaces allow to combine a multitude of such query strategies with different classifiers, thereby facilitating a quick mix and match, and enabling a rapid development of both active learning experiments and applications. To make various classifiers accessible in a consistent way, it integrates several well-known machine learning libraries, namely, scikit-learn, PyTorch, and huggingface transformers -- for which the latter integrations are available as optionally installable extensions. The library is available under the MIT License at this https URL
Comments: preprint
Subjects: Machine Learning (cs.LG); Computation and Language (cs.CL)
Cite as: arXiv:2107.10314 [cs.LG]
  (or arXiv:2107.10314v1 [cs.LG] for this version)

Submission history

From: Christopher Schröder [view email]
[v1] Wed, 21 Jul 2021 19:23:56 GMT (65kb,D)

Link back to: arXiv, form interface, contact.