From 5caed073eedd8efe5395f022376d5f340376fa4e Mon Sep 17 00:00:00 2001
From: Zach <zanussbaum@gmail.com>
Date: Sun, 9 Apr 2023 15:12:49 +0000
Subject: [PATCH] fix: topic model for embeddings

---
 build_map.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/build_map.py b/build_map.py
index 400dbb9b..53741077 100644
--- 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,)
\ No newline at end of file