model: graph_conv-relation_attention_mlp_h128_l3_edge_prediction | (graph_conv-relation_attention_mlp_h128_l3) | WandB: y95cc4qi

README.md

@@ -39,7 +39,7 @@ optionally incorporating relation types for relation-aware prediction.
   - Type: `relation_attention_mlp`
   - Relation-Aware: ✓
 
-**Training Date**: 2025-12-24
+**Training Date**: 2025-12-29
 
 For detailed experiment and training settings see this repository's `config.json` file.
 
@@ -47,15 +47,17 @@ For detailed experiment and training settings see this repository's `config.json
 
 | Metric | Value |
 |--------|-------|
-| Validation AUC | 0.8325 |
-| Test AUC | 0.8324 |
-| Validation AP | 0.8346 |
-| Test AP | 0.8345 |
+| Validation relation-weighted AUC | 0.8708 |
+| Test relation-weighted AUC | 0.8707 |
+| Validation AUC | 0.8332 |
+| Test AUC | 0.8328 |
+| Validation AP | 0.8351 |
+| Test AP | 0.8348 |
 
 
 ## Links
 
-- 📊 [W&B Run](https://wandb.ai/napistu/napistu-experiments/runs/x7m631ns)
+- 📊 [W&B Run](https://wandb.ai/napistu/napistu-experiments/runs/y95cc4qi)
 - 🌐 [Napistu](https://napistu.com)
 - 💻 [GitHub Repository](https://github.com/napistu/Napistu-Torch)
 - 📖 [Read the Docs](https://napistu-torch.readthedocs.io/en/latest)
@@ -71,7 +73,7 @@ To reproduce the environment used for training, run the following commands:
 pip install torch==2.8.0
 pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/2.8.0+cpu.html
 pip install 'napistu==0.8.5'
-pip install 'napistu-torch[pyg,lightning]==0.3.2'
+pip install 'napistu-torch[pyg,lightning]==0.3.4'
 ```
 
 ### 2. Setup Data Store