Machine learning backend

You can easily connect your favorite machine learning framework with Label Studio by using Heartex SDK.

That gives you the opportunities to use:

Tutorials

Quickstart

Here is a quick example tutorial on how to do that with simple text classification:

  1. Clone repo
    git clone https://github.com/heartexlabs/label-studio
  1. Create new ML backend
    label-studio-ml init my_ml_backend --script label-studio/ml/examples/simple_text_classifier.py
  1. Start ML backend server
    label-studio-ml start my_ml_backend
  1. Run Label Studio connecting it to the running ML backend:
    label-studio start text_classification_project --init --template text_sentiment --ml-backend-url http://localhost:9090

Create your own ML backend

Check examples in label-studio/ml/examples directory.