pub_soft/dgl
3
0
Fork 0
mirror of https://github.com/dmlc/dgl.git synced 2026-06-04 19:44:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

[DOC format]Update 1_introduction.py (#5748)

Browse Source 
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
This commit is contained in:
gmsft
2023-05-30 15:37:58 +08:00
committed by GitHub
parent 8a3256d8ff
commit 5f9e45d725
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tutorials/blitz/1_introduction.py
View File

@@ -64,7 +64,7 @@ import torch.nn.functional as F
dataset = dgl.data.CoraGraphDataset()
print("Number of categories:", dataset.num_classes)
print(f"Number of categories: {dataset.num_classes}")
######################################################################
@@ -185,9 +185,7 @@ def train(g, model):
if e % 5 == 0:
print(
"In epoch {}, loss: {:.3f}, val acc: {:.3f} (best {:.3f}), test acc: {:.3f} (best {:.3f})".format(
e, loss, val_acc, best_val_acc, test_acc, best_test_acc
)
f"In epoch {e}, loss: {loss:.3f}, val acc: {val_acc:.3f} (best {best_val_acc:.3f}), test acc: {test_acc:.3f} (best {best_test_acc:.3f})"
)