Skip to content

Commit a119c23

Browse files
authored
Correct the python sssp result by setting generate_eid to false in load_p2p_network dataset (alibaba#2762)
Fixes alibaba#2735
1 parent e4bb069 commit a119c23

File tree

5 files changed

+9
-7
lines changed

5 files changed

+9
-7
lines changed

docs/analytical_engine/tutorial_dev_algo_python.rst

+2-2
Original file line numberDiff line numberDiff line change
@@ -201,7 +201,7 @@ To run your own algorithms, you may trigger it in place where you defined it.
201201
import graphscope
202202
from graphscope.dataset import load_p2p_network
203203
204-
g = load_p2p_network()
204+
g = load_p2p_network(generate_eid=False)
205205
206206
# load my algorithm
207207
my_app = SSSP_Pregel()
@@ -227,4 +227,4 @@ Later, you can load your own algorithm from the gar package.
227227
my_app = load_app("/tmp/my_sssp_pregel.gar")
228228
229229
# run my algorithm over a graph and get the result.
230-
ret = my_app(g, src="6")
230+
ret = my_app(g, src="6")

docs/zh/analytics_engine.rst

+1-1
Original file line numberDiff line numberDiff line change
@@ -319,7 +319,7 @@ GraphScope 支持用户在自定义算法中通过 :code:`context.math` 上的
319319
import graphscope
320320
from graphscope.dataset import load_p2p_network
321321
322-
g = load_p2p_network()
322+
g = load_p2p_network(generate_eid=False)
323323
324324
# 加载自己的算法
325325
my_app = SSSP_Pregel()

python/graphscope/dataset/p2p_network.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
from graphscope.dataset.io_utils import download_file
2424

2525

26-
def load_p2p_network(sess=None, prefix=None, directed=False):
26+
def load_p2p_network(sess=None, prefix=None, directed=False, generate_eid=True):
2727
"""Load p2p graph.
2828
A peer-to-peer dataset derived from Gnutella peer-to-peer network, August 31 2002,
2929
with generated data on vertices and edges. See more details here:
@@ -38,6 +38,8 @@ def load_p2p_network(sess=None, prefix=None, directed=False):
3838
source URL. Defaults to None.
3939
directed (bool, optional): Determine to load a directed or undirected graph.
4040
Defaults to True.
41+
generate_eid (bool, optional): True will generate id for each edge as the first property.
42+
Defaults to True.
4143
4244
Returns:
4345
:class:`graphscope.framework.graph.GraphDAGNode`:
@@ -76,7 +78,7 @@ def load_p2p_network(sess=None, prefix=None, directed=False):
7678
if sess is None:
7779
sess = get_default_session()
7880

79-
graph = sess.g(directed=directed)
81+
graph = sess.g(directed=directed, generate_eid=generate_eid)
8082
graph = graph.add_vertices(
8183
os.path.join(prefix, "p2p-31_property_v_0"), "host"
8284
).add_edges(

tutorials/06_writing_your_own_algorithms.ipynb

+1-1
Original file line numberDiff line numberDiff line change
@@ -302,7 +302,7 @@
302302
"\n",
303303
"from graphscope.dataset import load_p2p_network\n",
304304
"\n",
305-
"graph = load_p2p_network(directed=False)"
305+
"graph = load_p2p_network(directed=False, generate_eid=False)"
306306
]
307307
},
308308
{

tutorials/zh/06_writing_your_own_algorithms.ipynb

+1-1
Original file line numberDiff line numberDiff line change
@@ -298,7 +298,7 @@
298298
"\n",
299299
"from graphscope.dataset import load_p2p_network\n",
300300
"\n",
301-
"graph = load_p2p_network(directed=False)"
301+
"graph = load_p2p_network(directed=False, generate_eid=False)"
302302
]
303303
},
304304
{

0 commit comments

Comments
 (0)