Machine Learning classique
Aperçu
Cette fonctionnalité est en Aperçu public.
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 démontre un flux de travail 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 un cluster GPU serverless. Elle couvre l'ingestion des données, le rééchantillonnage, l'entraînement de 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. |