gradsflow automl
✨ Enhancements and Bug fixes
Una biblioteca de capacitación de modelos de AutoML y PyTorch de código abierto
Documentos | Ejemplos
!!! atención GradsFlow está cambiando rápidamente. Habrá muchos cambios importantes hasta que lleguemos 0.1.0 . No dudes en enviarnos tu opinión creando un problema o únete a nuestro grupo de Slack.
GradsFlow es una biblioteca AutoML de código abierto basada en PyTorch. Nuestro objetivo es democratizar la IA y ponerla a disposición de todos.
Puede crear y entrenar automáticamente modelos de aprendizaje profundo para diferentes tareas en su computadora portátil o en un clúster remoto directamente desde su computadora portátil. Proporciona una API de entrenamiento de modelos potente y fácil de ampliar que se puede utilizar para entrenar casi cualquier modelo de PyTorch. Aunque GradsFlow tiene su propia API de entrenamiento de modelos, también es compatible con PyTorch Lightning Flash para proporcionar funciones más completas en diferentes tareas.
!!! info ¡Gradsflow está diseñado tanto para principiantes como para expertos ! AutoTasks proporciona AutoML sin código, mientras que Model and Tuner proporciona entrenamiento de modelos personalizado y optimización de hiperparámetros.
Recomendado :
El método recomendado para instalar gradsflow es con pip de PyPI o con conda del canal conda-forge.
con pepita
pip install -U gradsflowcon conda
conda install -c conda-forge gradsflowÚltimo (inestable):
También puedes instalar la última versión de última generación (podría ser inestable) de gradsflow , si te sientes lo suficientemente motivado, de la siguiente manera:
pip install git+https://github.com/gradsflow/gradsflow@main ¿Es usted principiante o no tiene experiencia en aprendizaje automático? Esta sección es para ti. Gradsflow AutoTask proporciona capacitación y creación automática de modelos en varias tareas diferentes, incluido el reconocimiento de imágenes, el análisis de sentimientos, el resumen de texto y más por venir.
Tuner proporciona una API simplificada para pasar del entrenamiento de modelos a la optimización de hiperparámetros.
gradsflow.core : Core define los componentes básicos de las tareas de AutoML.
gradsflow.autotasks : AutoTasks define diferentes tareas de ML/DL proporcionadas por la API de gradsflow AutoML.
gradsflow.model : GradsFlow Model proporciona una API de capacitación de modelos simple pero personalizable. Puede entrenar cualquier modelo de PyTorch usando model.fit(...) y es fácilmente personalizable para tareas más complejas.
gradsflow.tuner : búsqueda de hiperparámetros de AutoModel con cambios mínimos de código.
? Consulte ejemplos de cuadernos para obtener más información.
? Patrocinador en ko-fi
? ¿Necesitas apoyo? Contáctenos en [email protected]
Social : También puede seguirnos en Twitter @gradsflow y Linkedin para obtener las últimas actualizaciones.
Únase al grupo de Slack para chatear con nosotros.
Se aceptan contribuciones de cualquier tipo. Puede actualizar la documentación, agregar ejemplos, solucionar problemas identificados, agregar/solicitar una nueva función.
Para obtener más detalles, consulte las Pautas de contribución antes de contribuir.
Nos comprometemos a actuar e interactuar de manera que contribuyan a una comunidad abierta, acogedora, diversa, inclusiva y saludable.
Lea el Código de conducta completo del Pacto del Colaborador
GradsFlow se creó con la ayuda de increíbles proyectos de código abierto (incluidos, entre otros) Ray, PyTorch Lightning, HuggingFace Accelerate y TorchMetrics. Se inspira en múltiples proyectos Keras y FastAI.
