code: def main(params):
      logger.debug("Args: %s", str(params))
      logger.debug("Number of epochs %d", params["n_epochs"])
  
      svd = svd_training(params)
      # Save SVD model to the output directory for later use
      output_dir = os.environ.get("NNI_OUTPUT_DIR")
      surprise.dump.dump(
          os.path.join(output_dir, "model.dump"), algo=svd
      )
  
file_path: recommenders/recommenders/tuning/nni/svd_training.py
imports: import argparse
  import json
  import logging
  import os
  import numpy
  import pandas
  import nni
  import surprise
  import recommenders.evaluation.python_evaluation
  from recommenders.models.surprise.surprise_utils import predict, compute_ranking_predictions
label_desc: Use image node_id 3 for calling a global function with example usage: main(params) without return types