Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
acezen committed Jan 18, 2022
1 parent d2dd9a7 commit fde0151
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions python/graphscope/learning/graph.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,11 +79,11 @@ def __init__(self, session, graph, nodes=None, edges=None, gen_labels=None):
nodes=[("paper", features)]) # use "paper" node and features for training
edges=[("paper", "links", "paper")] # use the `paper->links->papers` edge type for training
gen_labels=[
# cuts "paper" nodes into 100 pieces, and uses random 75 pieces(75%) as traning dataset.
# cuts "paper" nodes into 100 pieces, and uses random 75 pieces(75%) as traning dataset
("train", "paper", 100, (0, 75)),
# cuts "paper" nodes into 100 pieces, and uses random 10 pieces(10%) as validation dataset.
# cuts "paper" nodes into 100 pieces, and uses random 10 pieces(10%) as validation dataset
("val", "paper", 100, (75, 85)),
# cuts "paper" nodes into 100 pieces, and uses random 15 pieces(15%) as test dataset.
# cuts "paper" nodes into 100 pieces, and uses random 15 pieces(15%) as test dataset
("test", "paper", 100, (85, 100)),
]
)
Expand All @@ -94,7 +94,6 @@ def __init__(self, session, graph, nodes=None, edges=None, gen_labels=None):
edges=[("paper", "links", "paper")] # use the `paper->links->papers` edge type for training
gen_labels=[
# cuts "paper" nodes into 100 pieces, and uses all pieces as traning dataset
# and just alias "paper" to "train".
("train", "paper", 100, (0, 100)),
"""
self._session = session
Expand Down

0 comments on commit fde0151

Please sign in to comment.