/ml-decorators

Decorators for Machine Learning libs

Primary LanguagePython

Machine Learning Decorators

Este repositório contém uma coleção de decoradores em Python, projetados para melhorar e simplificar a implementação de rotinas comuns em projetos de machine learning.

Características

  • Decoradores chamados de "brokers" com métodos para interação com Google Cloud Storage, BigQuery e o gerenciador de Credenciais.
  • Decoradores para monitoramento de treino de modelos.
  • Decoradores para gerenciamento e registro de experimentos.
  • Funcionalidades para facilitar a serialização e o carregamento de modelos.
  • Decorador para Pytorch com integração ao W&B
  • Decorador da Hugging Faces