def export_saved_model()

in retail/recommendation-system/bqml-scann/embeddings_lookup/lookup_creator.py [0:0]

• 10 lines of code
• 1 McCabe index (conditional complexity)

def export_saved_model(embedding_files_path, model_output_dir):
  print('Instantiating embedding lookup model...')
  embedding_lookup_model = EmbeddingLookup(embedding_files_path)
  print('Model is Instantiated.')
  
  signatures = {
    'serving_default': embedding_lookup_model.__call__.get_concrete_function(),
  }

  print('Exporting embedding lookup model as a SavedModel...')
  tf.saved_model.save(embedding_lookup_model, model_output_dir, signatures=signatures)
  print('SavedModel is exported.')