提交 174506bd 编写于 作者: W Webbley

remove contrib in hetergraph

上级 496d4d30
...@@ -58,8 +58,8 @@ Now, we can build a heterogenous graph by using PGL. ...@@ -58,8 +58,8 @@ Now, we can build a heterogenous graph by using PGL.
import paddle.fluid as fluid import paddle.fluid as fluid
import paddle.fluid.layers as fl import paddle.fluid.layers as fl
import pgl import pgl
from pgl.contrib import heter_graph from pgl import heter_graph
from pgl.contrib import heter_graph_wrapper from pgl import heter_graph_wrapper
g = heter_graph.HeterGraph(num_nodes=num_nodes, g = heter_graph.HeterGraph(num_nodes=num_nodes,
edges=edges, edges=edges,
...@@ -160,8 +160,3 @@ for epoch in range(30): ...@@ -160,8 +160,3 @@ for epoch in range(30):
train_loss = exe.run(fluid.default_main_program(), feed=feed_dict, fetch_list=[loss], return_numpy=True) train_loss = exe.run(fluid.default_main_program(), feed=feed_dict, fetch_list=[loss], return_numpy=True)
print('Epoch %d | Loss: %f'%(epoch, train_loss[0])) print('Epoch %d | Loss: %f'%(epoch, train_loss[0]))
``` ```
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册