From ff3753f32e93267e5f39e115d87c6205ffe10b94 Mon Sep 17 00:00:00 2001 From: Junwei Deng Date: Thu, 14 Nov 2019 21:40:57 -0500 Subject: [PATCH] change a typo num_samples -> num_sample so that make sampling mechanism work --- graphsage/model.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/graphsage/model.py b/graphsage/model.py index aeca282..1825b55 100644 --- a/graphsage/model.py +++ b/graphsage/model.py @@ -76,8 +76,8 @@ def run_cora(): agg2 = MeanAggregator(lambda nodes : enc1(nodes).t(), cuda=False) enc2 = Encoder(lambda nodes : enc1(nodes).t(), enc1.embed_dim, 128, adj_lists, agg2, base_model=enc1, gcn=True, cuda=False) - enc1.num_samples = 5 - enc2.num_samples = 5 + enc1.num_sample = 5 + enc2.num_sample = 5 graphsage = SupervisedGraphSage(7, enc2) # graphsage.cuda() @@ -148,8 +148,8 @@ def run_pubmed(): agg2 = MeanAggregator(lambda nodes : enc1(nodes).t(), cuda=False) enc2 = Encoder(lambda nodes : enc1(nodes).t(), enc1.embed_dim, 128, adj_lists, agg2, base_model=enc1, gcn=True, cuda=False) - enc1.num_samples = 10 - enc2.num_samples = 25 + enc1.num_sample = 10 + enc2.num_sample = 25 graphsage = SupervisedGraphSage(3, enc2) # graphsage.cuda()