Jerado's picture
Add BERTopic model
54efcc6 verified
raw
history blame contribute delete
582 Bytes
{
"calculate_probabilities": false,
"language": "english",
"low_memory": false,
"min_topic_size": 10,
"n_gram_range": [
1,
1
],
"nr_topics": null,
"seed_topic_list": [
[
"drug",
"cancer",
"drugs",
"doctor"
],
[
"windows",
"drive",
"dos",
"file"
],
[
"space",
"launch",
"orbit",
"lunar"
]
],
"top_n_words": 10,
"verbose": false,
"zeroshot_min_similarity": 0.7,
"zeroshot_topic_list": null,
"embedding_model": "sentence-transformers/all-MiniLM-L6-v2"
}