serjrd / label-studio-ml-backend

Configs and boilerplates for Label Studio's Machine Learning backend

Quickstart

Here is a quick example tutorial on how to run the ML backend with a simple text classifier:

0. Clone repo

   git clone https://github.com/heartexlabs/label-studio-ml-backend  

1. Setup environment

   cd label-studio-ml-backend
   pip install -e .
   cd label_studio_ml/examples
   pip install -r requirements.txt

2. Create new ML backend

   label-studio-ml init my_ml_backend --script label_studio_ml/examples/simple_text_classifier.py

3. Start ML backend server

   label-studio-ml start my_ml_backend

4. Run Label Studio connecting it to the running ML backend from the project settings page

Create your own ML backend

Check examples in label_studio_ml/examples directory.