Skip to content

Commit e75bc32

Browse files
Reduce verbose tqdm output in node2vec_movielens
1 parent 93d19ae commit e75bc32

File tree

3 files changed

+92
-396
lines changed

3 files changed

+92
-396
lines changed

examples/graph/ipynb/node2vec_movielens.ipynb

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -456,7 +456,12 @@
456456
" nodes = list(graph.nodes())\n",
457457
" for walk_iteration in range(num_walks):\n",
458458
" random.shuffle(nodes)\n",
459-
" for node in tqdm(nodes, desc=f\"Random walks iteration {walk_iteration + 1}\"):\n",
459+
" for node in tqdm(\n",
460+
" nodes,\n",
461+
" desc=f\"Random walks iteration {walk_iteration + 1}\",\n",
462+
" leave=False,\n",
463+
" mininterval=1.0,\n",
464+
" ):\n",
460465
" walk = [node]\n",
461466
" while len(walk) < num_steps:\n",
462467
" current = walk[-1]\n",
@@ -579,7 +584,12 @@
579584
" example_weights = defaultdict(int)\n",
580585
"\n",
581586
" # Iterate over all walks\n",
582-
" for sequence in tqdm(sequences, desc=\"Generating positive and negative examples\"):\n",
587+
" for sequence in tqdm(\n",
588+
" sequences,\n",
589+
" desc=\"Generating positive and negative examples\",\n",
590+
" leave=False,\n",
591+
" mininterval=1.0,\n",
592+
" ):\n",
583593
" # Use our manual skipgrams function\n",
584594
" pairs, labels = manual_skipgrams(\n",
585595
" sequence,\n",

0 commit comments

Comments
 (0)