trainer

Trainer à Gretel

Ce module est conçu pour fournir une interface simple pour aider les utilisateurs à former avec succès des modèles synthétiques sur des ensembles de données complexes avec des comptes de lignes et de colonnes élevés, et offre des fonctionnalités telles que la formation basée sur le SaaS Cloud et la parallélisation basée sur les GPU multi-GPU. Commencez gratuitement avec une clé API de Gretel.ai.

Fonctionnalité et fonctionnalités actuelles:

• Générateurs de données synthétiques pour les données de texte, tabulaires et de séries chronologiques avec les caractéristiques suivantes:
  • Équilibrez les ensembles de données ou stimulez une classe minoritaire en utilisant la génération de données conditionnelles.
  • Validation automatisée des données.
  • Rapports de qualité des données synthétiques.
  • Filtres de confidentialité et support de confidentialité différentielle facultatif.
• Plusieurs types de modèles pris en charge:
  • Le type de modèle Gretel-LSTM prend en charge le texte, le tabulaire, les séries chronologiques et la génération de données conditionnelles.
  • Le type de modèle Gretel-ACTGAN prend en charge la génération de données tabulaires et conditionnelles.
  • Synthèse du langage naturel Gretel-GPT basée sur une implémentation open source de GPT-3 (à venir bientôt).
  • Gretel-DGAN Multi-Variate Series basée sur Doppelganger (à venir bientôt).

Essayez-le maintenant!

Si vous souhaitez commencer rapidement à synthétiser des données avec Gretel.ai , cliquez simplement sur le bouton ci-dessous et suivez les exemples. Voir des exemples supplémentaires Python3 et Jupyter Notebook dans le dossier ./notebooks .

Rejoignez la discorde de la communauté des données synthétiques

Si vous souhaitez faire partie de la communauté des données synthétiques pour recevoir des annonces des dernières versions, poser des questions, suggérer de nouvelles fonctionnalités ou participer aux réunions de développement, veuillez rejoindre le serveur de communauté de données synthétiques!

Installer

Utilisation de pip :

pip install -U gretel-trainer

Start

1. Ajoutez votre clé API Gretel via la CLI Gretel.

Utilisez le client Gretel pour stocker votre clé API sur le disque. Cette étape est facultative, l'entraîneur provoquera une clé API à l'étape suivante.

gretel configure

2. Train ou affiner un modèle utilisant l'API Gretel

 from gretel_trainer import trainer

dataset = "https://gretel-public-website.s3-us-west-2.amazonaws.com/datasets/USAdultIncome5k.csv"

model = trainer . Trainer ()
model . train ( dataset )

3. Générez des données synthétiques!

 df = model . generate ()

Développement

Configurer l'environnement et installer les dépendances.

python -m venv venv
source venv/bin/activate
pip install -r requirements-dev.txt
pip install -e .

• Exécutez des tests via make test
• Exécuter le type de vérification (couverture limitée) via make type

Todos / feuille de route

• Activez la génération conditionnelle via l'interface SDK (prise en charge dans les ordinateurs portables actuellement).