code: def train_node2vec(graph, eval_set, args):
      """
      Train node2vec model
      """
      trainer = Node2vecModel(
          graph,
          embedding_dim=args.embedding_dim,
          walk_length=args.walk_length,
          p=args.p,
          q=args.q,
          num_walks=args.num_walks,
          eval_set=eval_set,
          eval_steps=1,
          device=args.device,
      )
  
      trainer.train(
          epochs=args.epochs,
          batch_size=args.batch_size,
          learning_rate=0.01,
      )
  
imports: import time
  from dgl.sampling import node2vec_random_walk
  from model import Node2vecModel
  from utils import load_graph, parse_arguments
label_desc: Use image node_id 2 for calling a global function with example usage: train_node2vec(graph, eval_set, args) without return types