learning early stop

learning/learning.py

@@ -146,7 +146,7 @@ def train_model(model, criterion, optimizer, device, train_loader, test_loader,
     """
 
     epochs_from_best = 0
-    early_stop_threshold = 80
+    early_stop_threshold = 10
 
     start_time = time.time()
     best_loss = sys.maxsize

post/validation.py

@@ -123,7 +123,7 @@ def ablation_results():
         graph_ids = pickle.load(open(f'../results/{run}/splits.p', 'rb'))
 
         acc, ranks_this  = decoy_test(model, decoys, edge_map, 32, 
-                                        test_graphlist=graph_ids['test'], 
+                                        test_graphlist=graph_ids['train'], 
                                         test_graph_path=graph_dir,
                                         shuffle=shuffle)
         ranks.extend(ranks_this)