未验证 提交 c2c94565 编写于 作者: Webbley's avatar Webbley 提交者: GitHub

Merge pull request #64 from PaddlePaddle/develop

Develop
pgl.contrib.heter\_graph module: Heterogenous Graph Storage pgl.heter\_graph module: Heterogenous Graph Storage
=============================== ===============================
.. automodule:: pgl.contrib.heter_graph .. automodule:: pgl.heter_graph
:members: :members:
:undoc-members: :undoc-members:
:show-inheritance: :show-inheritance:
pgl.contrib.heter\_graph\_wrapper module: Heterogenous Graph data holders for Paddle GNN. pgl.heter\_graph\_wrapper module: Heterogenous Graph data holders for Paddle GNN.
========================= =========================
.. automodule:: pgl.contrib.heter_graph_wrapper .. automodule:: pgl.heter_graph_wrapper
:members: :members:
:undoc-members: :undoc-members:
:show-inheritance: :show-inheritance:
...@@ -9,5 +9,5 @@ API Reference ...@@ -9,5 +9,5 @@ API Reference
pgl.data_loader pgl.data_loader
pgl.utils.paddle_helper pgl.utils.paddle_helper
pgl.utils.mp_reader pgl.utils.mp_reader
pgl.contrib.heter_graph pgl.heter_graph
pgl.contrib.heter_graph_wrapper pgl.heter_graph_wrapper
...@@ -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.
先完成此消息的编辑!
想要评论请 注册