提交 e85f3e88 编写于 作者: W Webbley

update

上级 703b864f
...@@ -55,21 +55,8 @@ class GINModel(object): ...@@ -55,21 +55,8 @@ class GINModel(object):
features_list.append(h) features_list.append(h)
pooled_h = pgl.layers.graph_pooling(self.gw, features_list[0], output = 0
self.pool_type)
output = fl.dropout(
pooled_h,
self.dropout_prob,
dropout_implementation="upscale_in_train")
output = fl.fc(output,
size=self.num_class,
act=None,
param_attr=fluid.ParamAttr(name="final_fc_0"))
for i, h in enumerate(features_list): for i, h in enumerate(features_list):
if i == 0:
continue
pooled_h = pgl.layers.graph_pooling(self.gw, h, self.pool_type) pooled_h = pgl.layers.graph_pooling(self.gw, h, self.pool_type)
drop_h = fl.dropout( drop_h = fl.dropout(
pooled_h, pooled_h,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册