model: graph_conv-relation_attention_h128_l3_edge_prediction | (graph_conv-relation_attention_h128_l3) | WandB: hh8kfhbv

wandb_run_info.yaml

@@ -1,9 +1,9 @@
 run_summaries:
-  _runtime: 128363
-  _step: 29859
-  _timestamp: 1767262307.039927
+  _runtime: 128375
+  _step: 29860
+  _timestamp: 1767300482.575149
   _wandb:
-    runtime: 128363
+    runtime: 128375
   avg_epoch_duration: 106.03179179318748
   corruption_error: 'supervision_edges shape mismatch: torch.Size([2, 6420899]) !=
     (2, 7154256)'
@@ -12,7 +12,7 @@ run_summaries:
   embeddings/embedding_norm_mean: 3.682695150375366
   embeddings/embedding_norm_median: 1.8914484977722168
   embeddings/embedding_norm_std: 4.762325763702393
-  epoch: 1499
+  epoch: 0
   epoch_duration_seconds: 78.35487675666809
   lr-AdamW: 0.0005
   score_distributions/head_type: relation_attention
@@ -29,12 +29,26 @@ run_summaries:
   score_distributions/rank_corr_with_dotprod: 0.513375740998943
   score_distributions/separation_cohens_d: 0.9815136790275574
   score_distributions/split: validation
+  test_ap: 0.7984310388565063
+  test_auc: 0.7990119457244873
+  test_auc_catalyst -> modified: 0.9728212952613832
+  test_auc_catalyst -> reactant: 0.9435023069381714
+  test_auc_inhibitor -> modified: 0.8967870473861694
+  test_auc_inhibitor -> reactant: 0.9266666769981384
+  test_auc_interactor -> interactor: 0.7878876328468323
+  test_auc_modifier -> modified: 0.9716925024986268
+  test_auc_other relation: 0.934612512588501
+  test_auc_reactant -> product: 0.944169282913208
+  test_auc_stimulator -> modified: 0.9381734728813172
+  test_auc_stimulator -> reactant: 0.9904575943946838
+  test_auc_stimulator -> stimulator: 0.9300100803375244
+  test_relation_weighted_auc: 0.8441495895385742
   time_per_epoch_avg: 106.03179179318748
   time_per_epoch_std: 720.338836282837
   total_epochs_completed: 1500
   total_train_time_minutes: 2652.899226737022
   train_loss: 0.03598431497812271
-  trainer/global_step: 29999
+  trainer/global_step: 0
   val_ap: 0.7972109317779541
   val_auc: 0.7983325719833374
   val_auc_catalyst -> modified: 0.971442461013794