Skip to content

GitLab

Unverified Commit b7ce4b6a authored by DominikaJedynak's avatar DominikaJedynak Committed by GitHub
Browse files

Tensorflow examples fix (#5433)

parent e102543b
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
examples/tensorflow/dgi/train.py
View file @ b7ce4b6a
...@@ -51,7 +51,7 @@ def main(args): ...@@ -51,7 +51,7 @@ def main(args):
test_mask = g.ndata["test_mask"] test_mask = g.ndata["test_mask"]
in_feats = features.shape[1] in_feats = features.shape[1]
n_classes = data.num_labels n_classes = data.num_labels
n_edges = data.graph.number_of_edges() n_edges = g.number_of_edges()
# add self loop # add self loop
if args.self_loop: if args.self_loop:
......
examples/tensorflow/gat/train.py
View file @ b7ce4b6a
...@@ -67,7 +67,7 @@ def main(args): ...@@ -67,7 +67,7 @@ def main(args):
test_mask = g.ndata["test_mask"] test_mask = g.ndata["test_mask"]
num_feats = features.shape[1] num_feats = features.shape[1]
n_classes = data.num_labels n_classes = data.num_labels
n_edges = data.graph.number_of_edges() n_edges = g.number_of_edges()
print( print(
"""----Data statistics------' """----Data statistics------'
#Edges %d #Edges %d
......
examples/tensorflow/gcn/train.py
View file @ b7ce4b6a
...@@ -44,7 +44,7 @@ def main(args): ...@@ -44,7 +44,7 @@ def main(args):
test_mask = g.ndata["test_mask"] test_mask = g.ndata["test_mask"]
in_feats = features.shape[1] in_feats = features.shape[1]
n_classes = data.num_labels n_classes = data.num_labels
n_edges = data.graph.number_of_edges() n_edges = g.number_of_edges()
print( print(
"""----Data statistics------' """----Data statistics------'
#Edges %d #Edges %d
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment