How to categorize text content with custom classes using Python?

How to categorize text content with custom classes using Python?

In this tutorial, you will learn how to use Custom Text Classification API in 5 minutes with Python. Eden AI provides an easy and developer-friendly API that allows you to classify text contents in custom classes.

What is Custom Text Classification API?

Text classification is a natural language processing (NLP) task that involves assigning predefined categories or labels to text documents based on their content. Unlike pre-trained models offered by generic text classification APIs, a Custom Text Classification API enables developers to build models from scratch or fine-tune existing models based on their unique datasets and requirements.

Custom Text Classification API result on Eden AI

With a Custom Text Classification API, developers have more control over the training process and can create models that are optimized for their specific domain or industry. They can define their own categories or labels and train the model using their annotated data. This flexibility allows for more accurate and specialized text classification results.

Get Started with Custom Text Classification API

The first step is to install Python's requests package, that will allow you to call Eden AI API.

Next, you'll need to install Python's JSON package to be able to read and print the result of the API request.

How to classify text with custom classes with Python

You are now ready to process your file into Eden AI Custom Text Classification API.

1. Get an Custom Text Classification API Key on Eden AI

To perform Custom Text Classification, you'll need to create an account on Eden AI for free. Then, you will be able to get your API key directly from the homepage with free credits offered by Eden AI.

2. Let’s classify text content with custom classify

Now that you have imported packages on Python and got your API key, you will be able to classify text contents with custom classes. With Eden AI, you can choose from a wide range of different engines you want for Custom Text Classification. You can see the list of Custom Text Classification providers available on Eden AI on our documentation.

Here is the Python script you need to write on your notebook:

For example, we called two different Custom Text Classification engines. Eden AI API will then return in its JSON response results of those providers.

Once the request is done, you will be able to get the result with this print:

Here is an example of a result for Custom Text Classification task:

Benefits of using Custom Text Classification API with Eden AI

Using Custom Text Classification with Eden AI API is quick and easy.

Multiple AIs in one API - Eden AI
Multiple AIs in one API - Eden AI

Save time and cost

We offer a unified API for all providers: simple and standard to use, with a quick switch between providers and an access to the specific features of each provider.

Easy to integrate

The JSON output format is the same for all suppliers thanks to Eden AI's standardisation work. The response elements are also standardised thanks to Eden AI's powerful matching algorithms.


With Eden AI you have the possibility to integrate a third party platform: we can quickly develop connectors. To go further and customize your Custom Text Classification request with specific parameters, check out our documentation.

Related Posts

Try Eden AI for free.

You can directly start building now. If you have any questions, feel free to schedule a call with us!

Get startedContact sales