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

Typo in maxpool name.

Browse Source
This commit is contained in:
William L Hamilton 2017-11-06 11:43:06 -08:00 committed by GitHub
parent 326710993f
commit b3af4dff3f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
graphsage/supervised_train.py
View File

@ -212,7 +212,7 @@ def train(train_data, test_data=None):
adj_info, adj_info,
minibatch.deg, minibatch.deg,
layer_infos=layer_infos, layer_infos=layer_infos,
aggregator_type="pool", aggregator_type="maxpool",
model_size=FLAGS.model_size, model_size=FLAGS.model_size,
sigmoid_loss = FLAGS.sigmoid, sigmoid_loss = FLAGS.sigmoid,
identity_dim = FLAGS.identity_dim, identity_dim = FLAGS.identity_dim,