Training and deploying a fine-tuned Cohere model.

In this section, we will walk through how you can start training a fine-tuning model for Classification with both the Web UI and the Python SDK.

Web UI

Creating a fine-tuned model for Classification with the Web UI consists of a few simple steps, which we’ll walk through now.

Choose the Classify Option

Go to the fine-tuning page and click on ‘Create a Classify model’.

Upload Your Data

Upload your custom dataset data by going to ‘Training data’ and clicking on the upload file button. Your data should be in csv or .jsonl format with exactly two columns—the first column consisting of the examples, and the second consisting of the labels.

You also have the option of uploading a validation dataset. This will not be used during training, but will be used for evaluating the model’s performance post-training. To upload a validation set, go to ‘Upload validation set (optional)’ and repeat the same steps you just went through with the training dataset. If you don’t upload a validation dataset, the platform will automatically set aside part of the training dataset to use for validation.

At this point in time, if there are labels in the training set with less than five unique examples, those labels will be removed.

Once done, click ‘Next’.

Preview Your Data

The preview window will show a few samples of your custom training dataset, and your validation dataset (if you uploaded it).

Toggle between the ‘Training’ and ‘Validation’ tabs to see a sample of your respective datasets.

At the bottom of this page, the distribution of labels in each respective dataset is shown.

If you are happy with how the samples look, click ‘Continue’.

Start Training

Now, everything is set for training to begin! Click ‘Start training’ to proceed.

Calling the Fine-tuned Model

Once your model completes training, you can call it by selecting ‘Try in Playground’ from the model list view:

or by selecting ‘Try in Playground’ from the model overview page:

Or, you can go to the classify playground and select your model from the drop down menu:

Python SDK

Text classification is one of the most common language understanding tasks. A lot of business use cases can be mapped to text classification. Examples include:

• Evaluating the tone and sentiment of an incoming customer message (e.g. classes: ‘positive’ and ‘negative’).
• Routing incoming customer messages to the appropriate agent (e.g. classes: ‘billing’, ‘tech support’, ‘other’).
• Evaluating if a user comment needs to be flagged for moderator attention (e.g. classes: ‘flag for moderation’, ‘neutral’).
• Evaluating which science topic a given piece of text is related to (e.g. classes: ‘biology’, ‘physics’). Since a given piece of text might be germane to more than one topic, this is an example of ‘multilabel’ classification, which is discussed in more detail at the end of this document.

Create a New Fine-tuned Model

In addition to using the Web UI for fine-tuning models, customers can also kick off fine-tuning jobs programmatically using the Cohere Python SDK. This can be useful for fine-tuning jobs that happen on a regular cadence, such as nightly jobs on newly-acquired data.

Using co.finetuning.create_finetuned_model(), you can create a fine-tuned model using either a single-label or multi-label dataset.

Examples

Here are some example code snippets for you to use.

Starting a single-label fine-tuning job

1
# create dataset
2
single_label_dataset = co.datasets.create(
3
    name="single-label-dataset",
4
    data=open("single_label_dataset.jsonl", "rb"),
5
    type="single-label-classification-finetune-input",
6
)
7
8
print(co.wait(single_label_dataset).dataset.validation_status)
9
                                              
10
# start the fine-tune job using this dataset
11
from cohere.finetuning.finetuning import BaseModel, FinetunedModel, Settings
12
13
single_label_finetune = co.finetuning.create_finetuned_model(
14
  request=FinetunedModel(
15
    name="single-label-finetune",
16
    settings=Settings(
17
      base_model=BaseModel(
18
        base_type="BASE_TYPE_CLASSIFICATION",
19
      ),
20
      dataset_id=single_label_dataset.id,
21
    ),
22
  ),
23
)
24
25
print(f"fine-tune ID: {single_label_finetune.finetuned_model.id}, fine-tune status: {single_label_finetune.finetuned_model.status}")

Starting a multi-label fine-tuning job

1
# create dataset
2
multi_label_dataset = co.datasets.create(
3
    name="multi-label-dataset",
4
    data=open("multi_label_dataset.jsonl", "rb"),
5
    type="multi-label-classification-finetune-input",
6
)
7
8
print(co.wait(multi_label_dataset).dataset.validation_status)
9
                                              
10
# start the fine-tune job using this dataset
11
from cohere.finetuning.finetuning import BaseModel, FinetunedModel, Settings
12
13
multi_label_finetune = co.finetuning.create_finetuned_model(
14
  request=FinetunedModel(
15
    name="multi-label-finetune",
16
    settings=Settings(
17
      base_model=BaseModel(
18
        base_type="BASE_TYPE_CLASSIFICATION",
19
      ),
20
      dataset_id=multi_label_dataset.id,
21
    ),
22
  ),
23
)
24
25
print(f"fine-tune ID: {multi_label_finetune.finetuned_model.id}, fine-tune status: {multi_label_finetune.finetuned_model.status}")

Calling a fine-tuned model

1
import cohere
2
3
co = cohere.Client('Your API key')
4
# get the custom model object (replace with your finetune name e.g. multi_label_finetune)
5
model_id = single_label_finetune.finetuned_model.id
6
7
8
response = co.classify(
9
    inputs=["classify this!"],
10
    model=model_id+"-ft"
11
)
12
13
print(response)

We can’t wait to see what you start building! Share your projects or find support on our Discord.