This experiment demonstrates the following steps of working with GNN within the GNN-AID framework:
- Loading the dataset and preparing the data.
- Building the GNN model.
- Training the GNN model.
- Saving and loading the model weights.
- Assessing the quality of the trained model.
train_gnn.py— script for training GNN.run_example.sh— script for running the experiment.README.md— description of the experiment.
Run:
bash run_example.sh| Metric | Value |
|---|---|
| F1 (macro, test) | ~0.83 |
| Accuracy (test) | ~0.85 |
The exact values may differ due to random weight initialization and model training process.