5. Train a model with the AML Dashboard

5.1. Background

The AML Dashboard is a web-based tool that allows users to interact with the AML framework.

This tutorial showcases the training process of an AML model using the AML Dashboard.

5.2. Prerequisites

Ensure you have installed the AML Dashboard using one of the following methods:

• Linux Installation

• Docker Image Installation

For more information, check the AML Dashboard Interfaces and AML Dashboard Usage sections.

5.3. Running the demo

To run the necessary components for training a model using the AML Dashboard, follow these steps:

5.3.1. Start the backend server

1. Navigate to the backend directory.

cd backend

2. Load the AML-IP environment.

source /AML-IP/install/setup.bash

3. Start the server:

python3 server.py

5.3.2. Start the Computing Node

1. Load the AML-IP environment.

source /AML-IP/install/setup.bash

2. Navigate to the backend directory.

cd backend

3. Start one or more computing nodes:

python3 computing.py

Each computing node will wait for job assignments and will collectively distribute the workload when multiple nodes are running.

5.3.3. Start the AML Dashboard

1. Navigate to the frontend/aml_dashboard directory.

cd frontend/aml_dashboard

2. Start the AML Dashboard:

npm run dev

3. Access the dashboard at http://localhost:5173/.

5.4. Training the Model

To train a model using the AML Dashboard, follow these steps:

1. Navigate to the Training tab on the AML Dashboard.

2. Specify the training parameters:

   • Number of parallel trainings (executions).

   • Number of iterations per execution.

   • Percentage of the dataset to distribute in each execution.

   • The target class when the classification is binary (this is the case when standard or custom datasets are used).

3. Configure the neural network parameters:

   • Number of layers.

   • Number of epochs.

   • Batch size.

   Note

   The target class previously set will also be used for the neural network training.

4. Optionally, upload an atomization file as a pre-trained model.

5. Click on the Train button in the AML Training Launcher to initiate the training process.

6. The training progress will be displayed, and the model status will update to Finished :) once the training is completed.