Overview
Agent AutoML is an AI agent-powered machine learning platform that automatically analyzes uploaded datasets and builds end-to-end data science workflows. The system identifies the type of problem (classification or regression), performs exploratory data analysis, generates insights, and trains multiple machine learning models automatically.
Problem to Solve
Building machine learning models typically requires significant manual effort, including data cleaning, exploratory analysis, feature understanding, model selection, and evaluation. This process becomes inefficient for non-expert users and slows down rapid experimentation.
Solution Approach
Developed an AI agent system that automatically analyzes uploaded datasets and determines the appropriate machine learning task such as classification or regression. The system then performs automated exploratory data analysis, including statistical summaries and visualizations to understand data distribution and relationships.
After analysis, the agent trains multiple machine learning models, evaluates their performance, and compares results in a structured table format. The best-performing model is identified, and all trained artifacts including models and reports can be downloaded by the user. The system is deployed using FastAPI for real-time interaction.
Key Technologies
- Python
- Machine Learning
- AutoML
- AI Agent
- Pandas
- Scikit-Learn
- Data Visualization
- FastAPI
.gif)