Add MLCube implementation for Graph Neural Network

[davidjurado]

MLCube for Graph Neural Network

MLCube™ GitHub repository. MLCube™ wiki.

Project setup

An important requirement is that you must have Docker installed.

# Create Python environment and install MLCube Docker runner 
virtualenv -p python3 ./env && source ./env/bin/activate && pip install pip==24.0 && pip install mlcube-docker
# Fetch the implementation from GitHub
git clone https://github.com/mlcommons/training && cd ./training
git fetch origin pull/762/head:feature/mlcube_graph_nn && git checkout feature/mlcube_graph_nn
cd ./graph_neural_network/mlcube

Inside the mlcube directory run the following command to check implemented tasks.

mlcube describe

MLCube tasks

Download dataset.

mlcube run --task=download_data -Pdocker.build_strategy=always

Process dataset.

mlcube run --task=process_data -Pdocker.build_strategy=always

Train GNN.

mlcube run --task=train -Pdocker.build_strategy=always

Execute the complete pipeline

You can execute the complete pipeline with one single command.

mlcube run --task=download_data,process_data,train -Pdocker.build_strategy=always

[github-actions[bot]]

MLCommons CLA bot All contributors have signed the MLCommons CLA ✍️ ✅

[ShriyaPalsamudram]

@drcanchi @LiSu can you please review these changes?

[LiSu]

@drcanchi @LiSu can you please review these changes?

LGTM, thanks! @hiwotadese @nv-rborkar could you please check this PR?