> For the complete documentation index, see [llms.txt](https://ubiai.gitbook.io/ubiai-documentation/llms.txt). Markdown versions of documentation pages are available by appending `.md` to page URLs; this page is available as [Markdown](https://ubiai.gitbook.io/ubiai-documentation/project-models.md).

# Project Models

UBIAI empowers users to create and train custom models for various natural language processing tasks, enabling efficient labeling. The models you can fine-tune includes NER, Relation Extraction, Span categorization and text classification.

## Named Entity Recognition

### Custom Models

You can select a custom trained Named Entity Recognition (NER) models tailored to your specific domain or task. The tool allows for easy annotation of labeled entities in text data. The process involves:

To use the model for auto-labeling, select it(only one model at a time can be selected) from the list of available models. The actual model can be trained from the Model Menu in the navigation bar, refer to the [Model-Assisted Labeling](/ubiai-documentation/model-assisted-labeling.md) section.

<figure><img src="https://3073024999-files.gitbook.io/~/files/v0/b/gitbook-x-prod.appspot.com/o/spaces%2F0hqV0hLtifjaqlfkGT7n%2Fuploads%2FlkSwECDiza2Lr1nv3L4K%2Fimage.png?alt=media&amp;token=337f5a8c-2b03-447c-8747-1b711198a999" alt=""><figcaption></figcaption></figure>

### LLM

Leverage Large Language Models (LLMs) for NER with zero-shot and few-shot learning approaches. Zero-shot learning enables the model to recognize entities without explicit training, while few-shot learning involves training the model on a small dataset with a few labeled examples.

To add an LLM model, simply click on "Add new model", by default OpenAI's GPT 3.5 Turbo will be added. You can configure the number of examples you would like to provide to GPT from 0 to 5. For more information, please refer to the [Zero-shot and Few-shot Labeling](/ubiai-documentation/zero-shot-and-few-shot-labeling.md) section.

<figure><img src="https://3073024999-files.gitbook.io/~/files/v0/b/gitbook-x-prod.appspot.com/o/spaces%2F0hqV0hLtifjaqlfkGT7n%2Fuploads%2FcnZUt1XCMU2UqTtKGucw%2Fimage.png?alt=media&amp;token=5d0dfe25-f74f-42da-aca1-2ff47199354e" alt=""><figcaption></figcaption></figure>

### Huggingface Models

Select from the thousands of pre-trained models in the Hugging Face (HF) Transformers library to auto-label your data.&#x20;

Simply press on "Add new model", input a name for your model and its HF URL as show below.

<figure><img src="https://3073024999-files.gitbook.io/~/files/v0/b/gitbook-x-prod.appspot.com/o/spaces%2F0hqV0hLtifjaqlfkGT7n%2Fuploads%2FrqbrMayC3UJBe6dngJUy%2Fimage.png?alt=media&amp;token=28ed0138-b89b-427e-804b-ea5667aacaa0" alt=""><figcaption></figcaption></figure>

## Relation Extraction

### Custom Models

Train relation extraction models to extract relationships between two entities in text data.&#x20;

To use the model for auto-labeling, select it (only one model at a time can be selected) from the list of available models. The actual model can be trained from the Model Menu in the navigation bar, refer to the [Model-Assisted Labeling](/ubiai-documentation/model-assisted-labeling.md) section.

### LLM

Coming soon...

## Span Categorizer

### Custom Models

Span Categorization involves labeling specific spans of text such as long sentences or paragraphs with predefined label. UBIAI facilitates training span categorizer models to identify and categorize spans.&#x20;

To use the model for auto-labeling, select it (only one model at a time can be selected) from the list of available models. The actual model can be trained from the Model Menu in the navigation bar, refer to the [Model-Assisted Labeling](/ubiai-documentation/model-assisted-labeling.md) section.

## Text Classification

### Custom Models

Train models to classify entire documents into predefined categories. To use the model for auto-labeling, select it (only one model at a time can be selected) from the list of available models. The actual model can be trained from the Model Menu in the navigation bar, refer to the [Model-Assisted Labeling](/ubiai-documentation/model-assisted-labeling.md) section.

### LLM

Leverage Large Language Models (LLMs) for text classfication with zero-shot and few-shot learning approaches. Zero-shot learning enables the model to recognize categories without explicit training, while few-shot learning involves training the model on a small dataset with a few labeled examples.

To add an LLM model, simply click on "Add new model", by default OpenAI's GPT 3.5 Turbo will be added. You can configure the number of examples you would like to provide to GPT from 0 to 5. For more information, please refer to the [Zero-shot and Few-shot Labeling](/ubiai-documentation/zero-shot-and-few-shot-labeling.md) section.