Machine Learning classique
Ces exemples exécutent des tâches classiques de Machine Learning sur AI Runtime. Ils montrent comment utiliser l’accélération GPU pour les algorithmes de ML traditionnels, les modèles de fondation tabulaires et les prévisions de séries temporelles, notamment la régression XGBoost, la prédiction zero-shot avec TabFM et les prévisions probabilistes avec GluonTS.
Exemple | Interface | Description |
|---|---|---|
Notebook | Ce notebook démontre comment entraîner un modèle de régression XGBoost sur un seul GPU. XGBoost peut bénéficier considérablement de l'accélération GPU pour les grands datasets. | |
Notebook | Ce Notebook illustre l’optimisation des hyperparamètres d’un modèle de classification XGBoost sur un seul GPU à l’aide d’Optuna et du suivi des Experimentation avec MLflow. | |
Notebook | Ce notebook démontre la classification et la régression zero-shot avec le modèle de fondation tabulaire TabFM de Google. TabFM utilise l’apprentissage en contexte pour effectuer des prédictions en une seule passe avant, sans affinement ni entraînement spécifique au dataset. | |
Notebook | Ce Notebook présente un workflow de bout en bout pour la prévision probabiliste de séries chronologiques de données de consommation d'électricité avec le modèle DeepAR de GluonTS sur l'environnement d'exécution de l'IA Runtime. Elle couvre l'ingestion des données, le rééchantillonnage, l'entraînement des modèles, la prédiction, la visualisation et l'évaluation. | |
CLI | Entraînez un classificateur XGBoost sur le dataset Forest CoverType avec un seul GPU A10, consignez les métriques dans MLflow, puis enregistrez le modèle dans Unity Catalog. | |
CLI | Utilisez Ray Tune et ASHA pour rechercher les hyperparamètres de XGBoost sur quatre GPU A10, consignez la meilleure configuration dans MLflow, et enregistrez le modèle dans Unity Catalog. | |
CLI | Entraînez un modèle GraphSAGE sur un GPU A10 pour classifier des documents de recherche dans le Graphe de citations ogbn-arxiv. | |
CLI | Utilisez Chronos-2 pour générer et évaluer des prévisions glissantes de prix de l'électricité à un jour sur un GPU A10. | |
Détection d'anomalies de séries chronologiques avec Kumo-Anomaly | CLI | Affinez NVIDIA Kumo-Anomaly sur un GPU A10 et comparez les modèles de base et affinés sur des données de détection d'anomalies retenues. |