def build()

in retail/recommendation-system/bqml-scann/index_builder/builder/indexer.py [0:0]


def build(embedding_files_pattern, output_dir, num_leaves=None):
  print("Indexer started...")
  tokens, embeddings = load_embeddings(embedding_files_pattern)
  index = build_index(embeddings, num_leaves)
  save_index(index, tokens, output_dir)
  print("Indexer finished.")