How to fix random seed for BERTopic?

user_3935797 0
Viewed 428

I'd like to fix the random seed from BERTopic library to get reproducible results. Looking at the code of BERTopic I see it uses numpy. Will using np.random.seed(123) be enough? or do I also need to other libraries as random or pytorch as in this question.

bert-language-model python
1 Answers

you can fix the random seed

from bertopic import BERTopic
from umap import UMAP

umap_model = UMAP(random_state=42)

topic_model = BERTopic(umap_model=umap_model)
user_171394070
Related