From 5f40ef2bce24b6396d5fe531ca1c5a61481c1fcb Mon Sep 17 00:00:00 2001 From: kiritofeng Date: Wed, 25 Jan 2023 14:06:11 -0500 Subject: [PATCH] Tutorial14: replace `transforms.AddTrainValTestMask` with `transforms.RandomNodeSplit` From `pytorch_geometric` v2.0.0 changelog: > `transforms.AddTrainValTestMask` has been replaced in favour of `transforms.RandomNodeSplit` --- Tutorial14/Tutorial14.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Tutorial14/Tutorial14.ipynb b/Tutorial14/Tutorial14.ipynb index 0698219..5ffd764 100644 --- a/Tutorial14/Tutorial14.ipynb +++ b/Tutorial14/Tutorial14.ipynb @@ -332,7 +332,7 @@ "source": [ "name = 'Cora'\n", "transform = transforms.Compose([\n", - " transforms.AddTrainValTestMask('train_rest', num_val=500, num_test=500),\n", + " transforms.RandomNodeSplit('train_rest', num_val=500, num_test=500),\n", " transforms.TargetIndegree(),\n", "])\n", "cora = datasets.Planetoid('./data', name, pre_transform=transforms.NormalizeFeatures(), transform=transform)"