WecoAI/aideml

O que é?

Recursos principais

• 01Descreva a tarefa em linguagem natural, informando um diretório de dataset, um objetivo e uma métrica de avaliação.
• 02Percorra uma árvore de scripts Python gerados, aprimorando ou depurando candidatos com base no retorno da métrica.
• 03Oferece uma CLI, uma API Python e uma imagem Docker.
• 04Inclui um visualizador HTML da árvore de soluções e uma interface web em Streamlit.
• 05Funciona com OpenAI, Anthropic, Gemini, OpenRouter e modelos locais compatíveis com OpenAI.

Casos de uso

• Criar um protótipo de modelo de ML básico para um dataset tabular a partir de um objetivo descrito em linguagem natural.
• Reproduzir ou ampliar os experimentos de busca em árvore do AIDE em pesquisas.
• Testar heurísticas de busca, avaliadores ou backends de LLM alternativos para agentes de ML.
• Executar uma pipeline totalmente local usando um modelo servido pelo Ollama.

Início rápido

1. 1

   Instale o pacote pelo PyPI

   pip install -U aideml
2. 2

   Configure sua chave de API do LLM

   export OPENAI_API_KEY=<your-key>
3. 3

   Execute uma otimização na tarefa de exemplo

   aide data_dir="example_tasks/house_prices" goal="Predict the sales price for each house" eval="RMSE between log-prices"

Antes de usar

• !As configurações padrão chamam APIs externas de LLM, então você precisa de uma chave de API e pagará pelo uso do modelo, a menos que execute um modelo local.
• !Execuções totalmente locais exigem um servidor local compatível com OpenAI e geralmente apresentam uma pequena queda de desempenho em comparação com modelos hospedados.

Em coleções

Tópicos

Perguntas frequentes

O que é aideml?

Um agente LLM de código aberto que cria, executa, depura e aprimora código de machine learning por meio de busca em árvore até otimizar uma métrica definida pelo usuário. AIDE ML é a implementação de referência de código aberto do algoritmo AIDE, um agente de busca em árvore que trata cada script Python como um nó em uma árvore de soluções. Dado um diretório de dataset, um objetivo descrito em linguagem natural e uma métrica de avaliação, ele gera código, executa-o e usa o retorno da métrica para decidir quais candidatos aprimorar ou corrigir.

aideml é gratuito? Qual é a licença?

aideml é open source sob a licença MIT. Confira os termos da licença antes de usá-lo em produtos comerciais.

aideml ainda é mantido?

Manutenção ativa. O último commit de aideml foi em 16 de out. de 2026.

Para que posso usar aideml?

Criar um protótipo de modelo de ML básico para um dataset tabular a partir de um objetivo descrito em linguagem natural. Reproduzir ou ampliar os experimentos de busca em árvore do AIDE em pesquisas. Testar heurísticas de busca, avaliadores ou backends de LLM alternativos para agentes de ML. Executar uma pipeline totalmente local usando um modelo servido pelo Ollama.

Como instalar aideml?

Instale o pacote pelo PyPI: pip install -U aideml Configure sua chave de API do LLM: export OPENAI_API_KEY=<your-key> Execute uma otimização na tarefa de exemplo: aide data_dir="example_tasks/house_prices" goal="Predict the sales price for each house" eval="RMSE between log-prices"

Onde consigo aideml?

O código-fonte e as instruções de instalação estão no GitHub: https://github.com/WecoAI/aideml