GraphStorm CLI Examples

GraphStorm provides Command Line Interfaces (CLIs) for graph data preparation and GML model training and inference. To help users to quickly onboard GraphStorm CLIs, we provide a set of Jupyter notebooks as examples.

Note

All runnable Jupyter notebooks can be downloaded from the GraphStorm Github repository.

These notebooks all use the same ACM data as discussed in the User Your Own Data Tutorial. Users can follow the Notebook 0: Data Preparation to explore the details of ACM data preparation.

The CLI Notebook: Use GraphStorm CLI for Multi-task Learning provides an example that demonstrates how to run multi-task GNN model training and inference with GraphStorm CLIs. The training tasks include link prediction and node feature reconstruction.

• Use GraphStorm CLIs for Multi-task Learning