Skip to content

Commit ad02ebd

Browse files
committed
chore: change default walker for HALK with MUTAG
1 parent a118d2b commit ad02ebd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/mutag.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from pyrdf2vec import RDF2VecTransformer
99
from pyrdf2vec.embedders import Word2Vec
1010
from pyrdf2vec.graphs import KG
11-
from pyrdf2vec.walkers import RandomWalker
11+
from pyrdf2vec.walkers import HALKWalker
1212

1313
# Ensure the determinism of this script by initializing a pseudo-random number.
1414
RANDOM_STATE = 22
@@ -32,7 +32,7 @@
3232
# Extract all walks with a maximum depth of 2 for each entity using two
3333
# processes and use a random state to ensure that the same walks are
3434
# generated for the entities.
35-
walkers=[RandomWalker(2, None, n_jobs=1, random_state=RANDOM_STATE)],
35+
walkers=[HALKWalker(2, None, n_jobs=2, random_state=RANDOM_STATE)],
3636
verbose=1,
3737
).fit_transform(
3838
KG(

0 commit comments

Comments
 (0)