fix: topic model for embeddings

1 year ago · 195f8a7d4e
parent 7807a80bbb
commit 195f8a7d4e
1 changed files with 2 additions and 0 deletions
--- a/build_map.py
+++ b/build_map.py
@ -49,4 +49,6 @@ atlas.map_embeddings(embeddings,
                     id_field="index",
                     name="Post Epoch 1 Embeddings",
                     colorable_fields=["source", "loss", "trained_on"],
+                     build_topic_model=True,
+                     topic_label_field="inputs",
                     reset_project_if_exists=True,)