From 4583602eec07d72de4c63516478c9ee2c3d4a602 Mon Sep 17 00:00:00 2001 From: William L Hamilton Date: Wed, 27 Dec 2017 16:27:30 -0600 Subject: [PATCH] Update model.py --- graphsage/model.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/graphsage/model.py b/graphsage/model.py index 9a08a1c..aeca282 100644 --- a/graphsage/model.py +++ b/graphsage/model.py @@ -12,6 +12,11 @@ from collections import defaultdict from graphsage.encoders import Encoder from graphsage.aggregators import MeanAggregator +""" +Simple supervised GraphSAGE model as well as examples running the model +on the Cora and Pubmed datasets. +""" + class SupervisedGraphSage(nn.Module): def __init__(self, num_classes, enc):