fix: topic model for embeddings

This commit is contained in:
Zach 2023-04-09 15:12:49 +00:00
parent c11ea7dc31
commit 5caed073ee

View File

@ -49,4 +49,6 @@ atlas.map_embeddings(embeddings,
id_field="index", id_field="index",
name="Post Epoch 1 Embeddings", name="Post Epoch 1 Embeddings",
colorable_fields=["source", "loss", "trained_on"], colorable_fields=["source", "loss", "trained_on"],
build_topic_model=True,
topic_label_field="inputs",
reset_project_if_exists=True,) reset_project_if_exists=True,)