gradsflow automl
✨ Enhancements and Bug fixes
Uma biblioteca de treinamento de modelo AutoML e PyTorch de código aberto
Documentos | Exemplos
!!! atenção GradsFlow está mudando rapidamente. Haverá muitas mudanças importantes até chegarmos 0.1.0 . Sinta-se à vontade para dar seu feedback criando um problema ou junte-se ao nosso grupo no Slack.
GradsFlow é uma biblioteca AutoML de código aberto baseada em PyTorch. Nosso objetivo é democratizar a IA e torná-la disponível para todos.
Ele pode construir e treinar automaticamente modelos de aprendizado profundo para diferentes tarefas em seu laptop ou em um cluster remoto diretamente de seu laptop. Ele fornece uma API de treinamento de modelo poderosa e fácil de estender que pode ser usada para treinar quase qualquer modelo PyTorch. Embora GradsFlow tenha sua própria API de treinamento de modelo, ele também oferece suporte a PyTorch Lightning Flash para fornecer recursos mais avançados em diferentes tarefas.
!!! info Gradsflow foi desenvolvido para iniciantes e especialistas ! AutoTasks fornece AutoML de código zero, enquanto Model and Tuner fornece treinamento de modelo personalizado e otimização de hiperparâmetros.
Recomendado :
O método recomendado para instalar gradsflow é com pip do PyPI ou com conda do canal conda-forge.
com pip
pip install -U gradsflowcom Conda
conda install -c conda-forge gradsflowMais recente (instável):
Você também pode instalar a versão mais recente (pode ser instável) do gradsflow , caso se sinta motivado o suficiente, da seguinte maneira:
pip install git+https://github.com/gradsflow/gradsflow@main Você é iniciante ou não tem experiência em aprendizado de máquina? Esta seção é para você. Gradsflow AutoTask fornece construção automática de modelos e treinamento em várias tarefas diferentes, incluindo reconhecimento de imagem, análise de sentimento, resumo de texto e muito mais.
Tuner fornece uma API simplificada para passar do treinamento de modelo para a otimização de hiperparâmetros.
gradsflow.core : Core define os blocos de construção das tarefas do AutoML.
gradsflow.autotasks : AutoTasks define diferentes tarefas de ML/DL que são fornecidas pela API gradsflow AutoML.
gradsflow.model : GradsFlow Model fornece uma API de treinamento de modelo simples, mas personalizável. Você pode treinar qualquer modelo PyTorch usando model.fit(...) e é facilmente personalizável para tarefas mais complexas.
gradsflow.tuner : pesquisa de hiperparâmetro do AutoModel com alterações mínimas de código.
? Confira exemplos de notebooks para saber mais.
? Patrocinador do ko-fi
? Você precisa de suporte? Contate-nos em [email protected]
Social : você também pode nos seguir no Twitter @gradsflow e no Linkedin para obter as atualizações mais recentes.
Junte-se ao grupo Slack para conversar conosco.
Contribuições de qualquer tipo são bem-vindas. Você pode atualizar a documentação, adicionar exemplos, corrigir problemas identificados, adicionar/solicitar um novo recurso.
Para mais detalhes, verifique as Diretrizes de Contribuição antes de contribuir.
Comprometemo-nos a agir e interagir de forma a contribuir para uma comunidade aberta, acolhedora, diversificada, inclusiva e saudável.
Leia o Código de Conduta do Contributor Covenant completo
GradsFlow é construído com a ajuda de incríveis projetos de código aberto (incluindo, mas não limitado a) Ray, PyTorch Lightning, HuggingFace Accelerate, TorchMetrics. É inspirado em vários projetos Keras e FastAI.
