Model Training#

Understand how machine learning models can be trained from within Flyte, with an added advantage of orchestration benefits.

Diabetes Classification

Train an XGBoost model on the Pima Indians Diabetes Dataset.

House Price Regression

Use dynamic workflows to train a multiregion house price prediction model using XGBoost.

MNIST Classification

Train a neural network on MNIST with PyTorch and W&B

NLP Processing with Gensim

Word embedding and topic modelling on lee background corpus with Gensim

Forecast Sales Using Rossmann Store Sales

Forecast sales data with data-parallel distributed training using Horovod on Spark.