################
ML Training
################

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

.. panels::
    :header: text-center
    :column: col-lg-12 p-2

    .. link-button:: auto/case_studies/ml_training/pima_diabetes/index
       :type: ref
       :text: Diabetes Classification
       :classes: btn-block stretched-link
    ^^^^^^^^^^^^
    Train an XGBoost model on the Pima Indians Diabetes Dataset.

    ---

    .. link-button:: auto/case_studies/ml_training/house_price_prediction/index
       :type: ref
       :text: House Price Regression
       :classes: btn-block stretched-link
    ^^^^^^^^^^^^
    Use dynamic workflows to train a multiregion house price prediction model using XGBoost.

    ---

    .. link-button:: auto/case_studies/ml_training/mnist_classifier/index
       :type: ref
       :text: MNIST Classification
       :classes: btn-block stretched-link
    ^^^^^^^^^^^^
    Train a neural network on MNIST with PyTorch and W&B

    ---

    .. link-button:: auto/case_studies/ml_training/nlp_processing/index
       :type: ref
       :text: NLP Processing with Gensim
       :classes: btn-block stretched-link
    ^^^^^^^^^^^^
    Word embedding and topic modelling on lee background corpus with Gensim

    ---

    .. link-button:: auto/case_studies/ml_training/spark_horovod/index
       :type: ref
       :text: Forecast Sales Using Rossmann Store Sales Data with Horovod and Spark
       :classes: btn-block stretched-link
    ^^^^^^^^^^^^
    Data-parallel distributed training using Horovod on Spark.


.. toctree::
    :maxdepth: -1
    :caption: Contents
    :hidden:

    auto/case_studies/ml_training/pima_diabetes/index
    auto/case_studies/ml_training/house_price_prediction/index
    auto/case_studies/ml_training/mnist_classifier/index
    auto/case_studies/ml_training/nlp_processing/index
    auto/case_studies/ml_training/spark_horovod/index


.. TODO: write tutorials for data parallel training, distributed training, and single node training