model: graph_conv-distmult_h128_l3_edge_prediction | (graph_conv-distmult_h128_l3) | WandB: kjv3q37g

wandb_run_info.yaml

@@ -1,9 +1,9 @@
 run_summaries:
-  _runtime: 30211
-  _step: 6831
-  _timestamp: 1767034491.9456031
+  _runtime: 30218
+  _step: 6832
+  _timestamp: 1767155073.019701
   _wandb:
-    runtime: 30211
+    runtime: 30218
   avg_epoch_duration: 87.94588071403281
   corruption_error: 'supervision_edges shape mismatch: torch.Size([2, 6420899]) !=
     (2, 8621157)'
@@ -12,7 +12,7 @@ run_summaries:
   embeddings/embedding_norm_mean: 53.10097885131836
   embeddings/embedding_norm_median: 47.42254638671875
   embeddings/embedding_norm_std: 21.938392639160156
-  epoch: 342
+  epoch: 0
   epoch_duration_seconds: 74.15779519081116
   lr-AdamW: 0.0005
   score_distributions/head_type: distmult
@@ -29,12 +29,26 @@ run_summaries:
   score_distributions/rank_corr_with_dotprod: 0.4860309751053818
   score_distributions/separation_cohens_d: 1.604447841644287
   score_distributions/split: validation
+  test_ap: 0.8283302187919617
+  test_auc: 0.8278802037239075
+  test_auc_catalyst -> modified: 0.9635054469108582
+  test_auc_catalyst -> reactant: 0.9896195530891418
+  test_auc_inhibitor -> modified: 0.886361837387085
+  test_auc_inhibitor -> reactant: 0.9801979660987854
+  test_auc_interactor -> interactor: 0.8198896646499634
+  test_auc_modifier -> modified: 0.957679033279419
+  test_auc_other relation: 0.9750974774360656
+  test_auc_reactant -> product: 0.99563068151474
+  test_auc_stimulator -> modified: 0.922423779964447
+  test_auc_stimulator -> reactant: 0.9925699830055236
+  test_auc_stimulator -> stimulator: 0.90967059135437
+  test_relation_weighted_auc: 0.865027904510498
   time_per_epoch_avg: 87.94588071403281
   time_per_epoch_std: 13.457729559284251
   total_epochs_completed: 343
   total_train_time_minutes: 503.24063789844513
   train_loss: 0.03498705103993416
-  trainer/global_step: 6859
+  trainer/global_step: 0
   val_ap: 0.8281906843185425
   val_auc: 0.8276978731155396
   val_auc_catalyst -> modified: 0.961272656917572