Classic machine learning

Important

This feature is in Public Preview.

These notebooks run classic machine learning tasks on AI Runtime. They show how to use GPU acceleration for traditional ML algorithms and time series forecasting, including XGBoost regression and probabilistic forecasting with GluonTS.

Tutorial	Description
XGBoost model training	This notebook demonstrates how to train an XGBoost regression model on a single GPU. XGBoost can significantly benefit from GPU acceleration for large datasets.
Time series forecasting with GluonTS	This notebook demonstrates an end-to-end workflow for probabilistic time-series forecasting of electricity consumption data with GluonTS's DeepAR model on a serverless GPU cluster. It covers data ingestion, resampling, model training, prediction, visualization, and evaluation.