How to Build a Classification Model with Promptrepo?

You sort items into categories by hand in Google Sheets: food pages by nutrition category, tickets by team, feedback by theme. It takes time, and two people won't always pick the same label. A general AI model can do it, but it may answer with a label you don't use, and training one on your own examples usually takes code. With Promptrepo, you build a classification model from your examples and it answers with one of your own categories. You don't write any code, and you can call the model as an API from your app or website.

Let's start. Create your AI model and follow the steps below.

Step 1: Prepare your training data

  1. Web page title: The name of the web page
  2. Web page content: Content of the page
  3. Nutrition category: The correct category label (e.g., Red Meat, Beverages, Cheese, etc.)
Each row is one example with these columns. You can start with one, as below, and add the rest to its Google Sheet later.

Step 2: Creating the model in Promptrepo

Input with Document, Email and Text to choose from, with Text marked
  • Enter one example in the box: a web page title and its content, such as "Classic beef burger. A quarter pound patty with onions and cheese in a soft bun."
  • Click Done. AI fills in the Output on the right.
Enter Text with the example typed in the box, with the Done button marked
  • Under Editing input, click + Add input field if you need another input.
  • Click Done
Editing input on the left and the Output filled in by AI on the right, with the Done button marked
  • Under Editing output, name the output Nutrition category and select Enum in its list. Enter your categories in place of Option 1 and Option 2. Click Add Option for each one you have.
An output named Nutrition category with Enum selected and marked, Option 1, Option 2 and Add Option
Note: In the output, choosing Enum for nutrition category makes it a classification model
  • Click Done
Input on the left and Editing output on the right, with the Done button marked
  • Click Publish
Input and Output side by side, with the Publish button marked at the top
  • Enter the AI model name and click Publish Now
The AI model name box with Food classification typed in, with the Publish Now button marked
  • When it says Model is ready!, click Proceed.
The Model is ready! message, with the Proceed button marked

Step 3: Training & Testing the Model

  • Click PREVIEW
The top of the model page with EDIT, CODE and PREVIEW, with PREVIEW marked
  • Enter a food name and its description and click PROCEED.
  • The model should predict the correct category
  • Click ADD TO TRAINING DATA on a result you like, to teach the model with one more example
The Input and the Output the model gave, with the Add to training data button marked
To add many examples at once, on the model page, click on the edit icon on the Training data tab. Its Google Sheet will open and you can add rows there.
The Training data tab of Food classification-V1 with the example in a row, with the edit icon marked
Classification models are foundational for structuring data. Now, let's explore how to build an extraction model for retrieving structured data from unstructured text.

Summary

Sorting items into set categories by hand takes time, and the labels don't always match. With Promptrepo, you enter an example, choose Enum for the output and publish a classification model that answers with one of your own categories. Create your AI model and complete the steps above. If you have any questions, write to support.

Last updated: