huihun/graphsage
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update model.py

Browse Source
This commit is contained in:
William L Hamilton 2017-12-27 16:27:30 -06:00 committed by GitHub
parent 314f98be08
commit 4583602eec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
graphsage/model.py
View File

@ -12,6 +12,11 @@ from collections import defaultdict
from graphsage.encoders import Encoder from graphsage.encoders import Encoder
from graphsage.aggregators import MeanAggregator 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): class SupervisedGraphSage(nn.Module):
def __init__(self, num_classes, enc): def __init__(self, num_classes, enc):