Skip to main content
The Oumi agent is designed to help users without deep machine learning expertise build custom AI models quickly and efficiently. The Agent can guide you through the full model development lifecycle, including generating synthetic training data, fine-tuning open-source models, and evaluating model performance.

​
Using the agent

You can access the Oumi agent through the UI at every stage of the workflow for guidance and recommendations. The agent also works in the background to help ensure your setup remains complete and consistent, surfacing missing configurations, flagging potential issues, and highlighting critical steps as you progress through each phase.

​
Try it out

Suppose you want to build a fraud classification model from scratch. You can get started with a simple prompt like: 
I need to create a classification model to predict whether an e-commerce transaction is fraudulent. Can you guide me through the steps?
If you issue this prompt, the agent will guide you through each stage of the workflow, explaining its actions along the way and completing the necessary steps on your behalf.
Check out the Prompt Library for more examples of what you can do with the Oumi agent.
By continuously monitoring your progress and validating key inputs, the Oumi agent helps reduce errors, prevent misconfigurations, and keep your workflow on track, so you can focus on building and improving your model with confidence.