test different sizes for y_pred_pos and y_pred_neg

snap-stanford · Aug 14, 2022 · f5534d9 · f5534d9
1 parent 9a615fc
commit f5534d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ogb/linkproppred/evaluate.py b/ogb/linkproppred/evaluate.py
@@ -299,7 +299,7 @@ def _eval_rocauc(self, y_pred_pos, y_pred_neg, type_info):
     evaluator = Evaluator(name = 'ogbl-vessel')
     print(evaluator.expected_input_format)
     print(evaluator.expected_output_format)
-    y_pred_pos = torch.tensor(np.random.randn(1000,))
+    y_pred_pos = torch.tensor(np.random.randn(1500,))
     y_pred_neg = torch.tensor(np.random.randn(1000,))
     input_dict = {'y_pred_pos': y_pred_pos, 'y_pred_neg': y_pred_neg}
     result = evaluator.eval(input_dict)