Delete previous models, they can be found in their respective repositories.

topN.py

@@ -11,7 +11,7 @@ from deepsnap.hetero_gnn import HeteroSAGEConv
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')  # It defines whether to execute on cpu or gpu.
 constructor = heterograph_construction.DISNETConstructor(device=device)  # Graph constructor.
 edge = ('disorder', 'dis_dru_the', 'drug')  # Graph edge type to study.
-n = 20  # Number of new predictions.
+n = 200000  # Number of new predictions.
 
 """
 It filters the predictions so just new predictions are considered.