fix: topic model for embeddings

2024-10-01 01:06:10 -04:00 · 2023-04-09 15:12:49 +00:00 · 2023-04-09 15:12:49 +00:00 · 195f8a7d4e
commit 195f8a7d4e
parent 7807a80bbb
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,)