未验证 提交 34b71dba 编写于 作者: W Weiyue Su 提交者: GitHub

Merge pull request #2 from WeiyueSu/master

fix example
...@@ -97,28 +97,28 @@ def build_graph_model(graph_wrapper, num_class, k_hop, graphsage_type, ...@@ -97,28 +97,28 @@ def build_graph_model(graph_wrapper, num_class, k_hop, graphsage_type,
feature, feature,
hidden_size, hidden_size,
act="relu", act="relu",
name="graphsage_mean_%s % i") name="graphsage_mean_%s" % i)
elif graphsage_type == 'graphsage_meanpool': elif graphsage_type == 'graphsage_meanpool':
feature = graphsage_meanpool( feature = graphsage_meanpool(
graph_wrapper, graph_wrapper,
feature, feature,
hidden_size, hidden_size,
act="relu", act="relu",
name="graphsage_meanpool_%s % i") name="graphsage_meanpool_%s" % i)
elif graphsage_type == 'graphsage_maxpool': elif graphsage_type == 'graphsage_maxpool':
feature = graphsage_maxpool( feature = graphsage_maxpool(
graph_wrapper, graph_wrapper,
feature, feature,
hidden_size, hidden_size,
act="relu", act="relu",
name="graphsage_maxpool_%s % i") name="graphsage_maxpool_%s" % i)
elif graphsage_type == 'graphsage_lstm': elif graphsage_type == 'graphsage_lstm':
feature = graphsage_lstm( feature = graphsage_lstm(
graph_wrapper, graph_wrapper,
feature, feature,
hidden_size, hidden_size,
act="relu", act="relu",
name="graphsage_maxpool_%s % i") name="graphsage_maxpool_%s" % i)
else: else:
raise ValueError("graphsage type %s is not" raise ValueError("graphsage type %s is not"
" implemented" % graphsage_type) " implemented" % graphsage_type)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册